b10-transfer 0.3.10__tar.gz → 0.3.11__tar.gz

{b10_transfer-0.3.10 → b10_transfer-0.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: b10-transfer
-Version: 0.3.10
+Version: 0.3.11
 Summary: Distributed PyTorch file transfer for Baseten - Environment-aware, lock-free file transfer management
 License: MIT
 Keywords: pytorch,file-transfer,cache,machine-learning,inference

{b10_transfer-0.3.10 → b10_transfer-0.3.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "b10-transfer"
-version = "0.3.10"
+version = "0.3.11"
 description = "Distributed PyTorch file transfer for Baseten - Environment-aware, lock-free file transfer management"
 authors = ["Shounak Ray <shounak.noreply@baseten.co>", "Fred Liu <fred.liu.noreply@baseten.co>"]
 maintainers = ["Fred Liu <fred.liu.noreply@baseten.co>", "Shounak Ray <shounak.noreply@baseten.co>"]

{b10_transfer-0.3.10 → b10_transfer-0.3.11}/src/b10_transfer/__init__.py

@@ -9,7 +9,7 @@ from .constants import OperationStatus
 from .logging_utils import get_b10_logger
 
 # Version
-__version__ = "0.3.10"
+__version__ = "0.3.11"
 
 __all__ = [
     "CacheError",

{b10_transfer-0.3.10 → b10_transfer-0.3.11}/src/b10_transfer/cache.py

@@ -166,6 +166,7 @@ def load_compile_cache() -> OperationStatus:
     """
     with cache_operation("Load"):
         b10fs_dir, torch_dir, work_dir = _setup_cache_paths()
+        logger.info(f"TORCHINDUCTOR_CACHE_DIR IS {config.TORCH_CACHE_DIR}")
 
         cache_filename = get_cache_filename()
         final_file, _ = _get_cache_file_paths(cache_filename, b10fs_dir)

{b10_transfer-0.3.10 → b10_transfer-0.3.11}/src/b10_transfer/cache_cli.py

@@ -89,6 +89,8 @@ def main() -> None:
 
     logger = _setup_logging(cfg.loglevel)
 
+    logger.info(f"TORCHINDUCTOR_CACHE_DIR IS {config.TORCH_CACHE_DIR}")
+
     # 1) Preload any existing cache (non-fatal on error)
     try:
         if load_compile_cache() == OperationStatus.SUCCESS:
@@ -114,5 +116,6 @@ def main() -> None:
     except Exception as e:
         logger.exception("save_compile_cache() failed: %s", e)
         sys.exit(3)
+    logger.info(f"TORCHINDUCTOR_CACHE_DIR IS {config.TORCH_CACHE_DIR}")
 
     logger.info("vLLM automatic torch compile cache done.")