torchmonarch-nightly 2025.9.8__cp313-cp313-manylinux2014_x86_64.whl → 2025.9.9__cp313-cp313-manylinux2014_x86_64.whl

monarch/__init__.py

@@ -9,13 +9,6 @@
 from importlib import import_module as _import_module
 from typing import TYPE_CHECKING
 
-# Import before monarch to pre-load torch DSOs as, in exploded wheel flows,
-# our RPATHs won't correctly find them.
-try:
-    import torch  # noqa: F401
-except ImportError:
-    pass
-
 # submodules of monarch should not be imported in this
 # top-level file because it will cause them to get
 # loaded even if they are not actually being used.

monarch/_src/actor/bootstrap_main.py

@@ -17,12 +17,6 @@ import multiprocessing
 import os
 import sys
 
-# Import torch to avoid import-time races if a spawned actor tries to import torch.
-try:
-    import torch  # @manual  # noqa: F401
-except ImportError:
-    pass
-
 
 async def main():
     from monarch._rust_bindings.monarch_hyperactor.bootstrap import bootstrap_main
@@ -38,6 +32,7 @@ def invoke_main():
     global bootstrap_main
 
     # TODO: figure out what from worker_main.py we should reproduce here.
+
     from monarch._src.actor.telemetry import TracingForwarder  # noqa
 
     if os.environ.get("MONARCH_ERROR_DURING_BOOTSTRAP_FOR_TESTING") == "1":

monarch/_src/actor/pickle.py

@@ -8,15 +8,18 @@
 
 import io
 import pickle
+import sys
 from contextlib import contextmanager, ExitStack
 from typing import Any, Callable, Iterable, List, Tuple
 
 import cloudpickle
 
-try:
-    import torch  # @manual
-except ImportError:
-    torch = None
+
+def maybe_torch():
+    """
+    We have to do some special pickling if torch is loaded but not if it isn't loaded?
+    """
+    return sys.modules.get("torch")
 
 
 _orig_function_getstate = cloudpickle.cloudpickle._function_getstate
@@ -76,6 +79,7 @@ def flatten(obj: Any, filter: Callable[[Any], bool]) -> Tuple[List[Any], bytes]:
 
 def unflatten(data: bytes, values: Iterable[Any]) -> Any:
     with ExitStack() as stack:
+        torch = maybe_torch()
         if torch is not None:
             stack.enter_context(load_tensors_on_cpu())
             stack.enter_context(torch.utils._python_dispatch._disable_current_modes())
@@ -87,6 +91,8 @@ def unflatten(data: bytes, values: Iterable[Any]) -> Any:
 def load_tensors_on_cpu():
     # Ensure that any tensors load from CPU via monkeypatching how Storages are
     # loaded.
+    import torch
+
     old = torch.storage._load_from_bytes
     try:
         torch.storage._load_from_bytes = lambda b: torch.load(

{torchmonarch_nightly-2025.9.8.dist-info → torchmonarch_nightly-2025.9.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: torchmonarch-nightly
-Version: 2025.9.8
+Version: 2025.9.9
 Summary: Monarch: Single controller library
 Author: Meta
 Author-email: oncall+monarch@xmail.facebook.com

{torchmonarch_nightly-2025.9.8.dist-info → torchmonarch_nightly-2025.9.9.dist-info}/RECORD

@@ -1,5 +1,5 @@
-monarch/__init__.py,sha256=mgKiyD1kxky-1pvhMlNfF4VmxWnhi-FSYZNFzkW1BEM,7052
-monarch/_rust_bindings.so,sha256=DbwW1G8kuEeEnQJy954ZdQ__vgDFxo-NyOm6W4Nf2vE,61550376
+monarch/__init__.py,sha256=cH7tTOsGn-qoeBmSVv_67yHxxsQhMQD2aBScPPkNO7k,6870
+monarch/_rust_bindings.so,sha256=XMUUyzFMIuYh3vIu4OU025Vtyl25jGcwFw2hzYh58iY,61507624
 monarch/_testing.py,sha256=5BDMVA4hBMo780rsJ39vRmUZi6mTN8aYY7I9grJRjJ8,7841
 monarch/actor_mesh.py,sha256=VtPU9syi_vUdwDSJJ639Z4Y_EcWZUScyoj0lQ88RQPs,421
 monarch/bootstrap_main.py,sha256=39OZpNMrfvvNJf-iwuNzgslzYA_ItaRPHfXGn_V74N0,524
@@ -8,7 +8,7 @@ monarch/fetch.py,sha256=CssP25dMqyJnJAWoC41lwkMnSbvS-f2DL9PRbudJXfc,1704
 monarch/gradient_generator.py,sha256=b7PmoN_F3c5hQglfHeW_v5htYnePKvJGkzZN-tpHR4A,6396
 monarch/memory.py,sha256=ol86dBhFAJqg78iF25-BuK0wuwj1onR8FIioZ_B0gjw,1377
 monarch/mesh_controller.py,sha256=Y_26Cnmp72TccNbWdDQhq18j7de7pSw83E_fREJX9Yo,15372
-monarch/monarch_controller,sha256=npifJaDq61NfnwhKRxRkRLSCx3DiyNfTtQ3Pzh9aFNk,32549008
+monarch/monarch_controller,sha256=SD2c06R9ClraWw-Uwui05KragWtk9RmU89cLQ2IjXPE,30598480
 monarch/notebook.py,sha256=zu9MKDFKf1-rCM2TqFSRJjMBeiWuKcJSyUFLvoZRQzs,25949
 monarch/opaque_module.py,sha256=jCcg0DjbcEVXA9WNG0NhUzGteLHOJLTZEBvrIYJIAns,10436
 monarch/opaque_object.py,sha256=x1LoX6RIMGh4ux52xIfhPgoh6PhZHdkf9bMccHW3DW0,2808
@@ -27,14 +27,14 @@ monarch/_src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monarch/_src/actor/__init__.py,sha256=4iK3nzQZmEPe0HGNY70fABBenw3lCVVaaF0xddF5Fa0,235
 monarch/_src/actor/actor_mesh.py,sha256=P9b4CvxYeYfJecPxeEtohAasvGgzzElcvxh9chALeAA,40526
 monarch/_src/actor/allocator.py,sha256=UVGhrkPQMqPQp6vUngPI361s6yCEfZ0gfz8WTtG2om4,9392
-monarch/_src/actor/bootstrap_main.py,sha256=7T7ARumcHLZ5RI-k5jej9tBil0J7-BUSVFKwAZO2tJU,2413
+monarch/_src/actor/bootstrap_main.py,sha256=Dp9VVNILHxzmoqtiY5cBGZzvf0tTrIjD81eGdnmdpuY,2253
 monarch/_src/actor/device_utils.py,sha256=gBpl23wMjppVAEzzj8U9HyX-B7Bs2_3ftiMAkzUS4j4,577
 monarch/_src/actor/endpoint.py,sha256=ndIxeXfXKpW-nb-CtYsDpt2UjQwRIdNZfmSL218yHd0,10887
 monarch/_src/actor/event_loop.py,sha256=2i4fKIkemBzua_t47BqVa2roZ6fWB6sbmMFPNx2zKN0,2832
 monarch/_src/actor/future.py,sha256=idgqzU_B5qWfClIP5dTLapmBflWq5va-ujAzUbT1Asc,7490
 monarch/_src/actor/host_mesh.py,sha256=8SOkg_LhHuzLyhpwxT7Yw1_h8QrIlwfWhrSwHyAvfnk,5083
 monarch/_src/actor/logging.py,sha256=9aguohqCtvLVwWGTFM7o-rBptT26BjI2s6E5Of2imM4,3311
-monarch/_src/actor/pickle.py,sha256=FhdbAEsGrsD7f25bxF7HlROLm6j2TTvmToq8P1kyhB8,2913
+monarch/_src/actor/pickle.py,sha256=ZK2vzI7BJeZHMLh6vLdQFACqvSlyWldJO_kzDPTUTGQ,3061
 monarch/_src/actor/proc_mesh.py,sha256=lYrRMQNOGAdFXuFvc3lQ68xIS01YJWMkpi8qH5HHAHE,27791
 monarch/_src/actor/python_extension_methods.py,sha256=QujLWOQQbDdGCin8tZfDxyIwkM-Md4t9QtcTGTHOE_s,3493
 monarch/_src/actor/shape.py,sha256=PJqxpQEISHlxK8rrlKWpcNMEHiGxBbc6TsHcGZCOsyE,8472
@@ -186,9 +186,9 @@ tests/simulator/test_profiling.py,sha256=TGYCfzTLdkpIwnOuO6KApprmrgPIRQe60KRX3wk
 tests/simulator/test_simulator.py,sha256=LO8lA0ssY-OGEBL5ipEu74f97Y765TEwfUOv-DtIptM,14568
 tests/simulator/test_task.py,sha256=ipqBDuDAysuo1xOB9S5psaFvwe6VATD43IovCTSs0t4,2327
 tests/simulator/test_worker.py,sha256=QrWWIJ3HDgDLkBPRc2mwYPlOQoXQcj1qRfc0WUfKkFY,3507
-torchmonarch_nightly-2025.9.8.dist-info/licenses/LICENSE,sha256=e0Eotbf_rHOYPuEUlppIbvwy4SN98CZnl_hqwvbDA4Q,1530
-torchmonarch_nightly-2025.9.8.dist-info/METADATA,sha256=7hmFosBGSgVwKVtit03yWAUuQjHWLhBtgjPi2sM6Fpo,6473
-torchmonarch_nightly-2025.9.8.dist-info/WHEEL,sha256=OlISbtpDcfagPrLwG7WtpcZbPTUnoKPnwphA_26fNqE,104
-torchmonarch_nightly-2025.9.8.dist-info/entry_points.txt,sha256=60QVSpYVzkzS4iDOiLp0fsLxVp47X3J2l3v7W-59LMo,117
-torchmonarch_nightly-2025.9.8.dist-info/top_level.txt,sha256=E-ZssZzyM17glpVrh-S9--qJ-w9p2EjuYOuNw9tQ4Eg,33
-torchmonarch_nightly-2025.9.8.dist-info/RECORD,,
+torchmonarch_nightly-2025.9.9.dist-info/licenses/LICENSE,sha256=e0Eotbf_rHOYPuEUlppIbvwy4SN98CZnl_hqwvbDA4Q,1530
+torchmonarch_nightly-2025.9.9.dist-info/METADATA,sha256=SMfT1xHAuNlQkzHJZ4paq0GNL_PA_hZlLGReEAe9ZJU,6473
+torchmonarch_nightly-2025.9.9.dist-info/WHEEL,sha256=OlISbtpDcfagPrLwG7WtpcZbPTUnoKPnwphA_26fNqE,104
+torchmonarch_nightly-2025.9.9.dist-info/entry_points.txt,sha256=60QVSpYVzkzS4iDOiLp0fsLxVp47X3J2l3v7W-59LMo,117
+torchmonarch_nightly-2025.9.9.dist-info/top_level.txt,sha256=E-ZssZzyM17glpVrh-S9--qJ-w9p2EjuYOuNw9tQ4Eg,33
+torchmonarch_nightly-2025.9.9.dist-info/RECORD,,