fix(serialization): Add extra logging statements for pwrite

coreweave · Nov 25, 2024 · 9d30c2a · 9d30c2a
1 parent 5e5fe76
commit 9d30c2a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tensorizer/serialization.py b/tensorizer/serialization.py
@@ -3549,7 +3549,15 @@ def _pwrite_syscall(
                 )
                 if bytes_just_written == 0:
                     if requested_write_size == 0:
-                        logger.error("pwrite: Attempted to write 0 bytes")
+                        if logger.isEnabledFor(logging.ERROR):
+                            logger.error(
+                                (
+                                    "pwrite: Attempted to write 0 bytes,"
+                                    " original size: %d, original type: %s"
+                                ),
+                                self._buffer_size(data),
+                                repr(data.__class__.__name__),
+                            )
                         break
                     attempts += 1
                     logger.debug("pwrite: Retrying (attempt %d/3)", attempts)