torchmonarch-nightly 2025.7.31__cp313-cp313-manylinux2014_x86_64.whl → 2025.8.2__cp313-cp313-manylinux2014_x86_64.whl

monarch/_src/actor/proc_mesh.py

@@ -65,15 +65,19 @@ from monarch._src.actor.shape import MeshTrait
 
 HAS_TENSOR_ENGINE = False
 try:
+    # Torch is needed for tensor engine
+    import torch  # @manual
+
+    # Confirm that rust bindings were built with tensor engine enabled
     from monarch._rust_bindings.rdma import (  # type: ignore[import]
         _RdmaBuffer,
         _RdmaManager,
     )
 
     # type: ignore[16]
-    HAS_TENSOR_ENGINE = _RdmaBuffer.rdma_supported()
+    HAS_TENSOR_ENGINE = torch.cuda.is_available()
 except ImportError:
-    logging.warning("RDMA is not available on this platform")
+    logging.warning("Tensor engine is not available on this platform")
 
 
 if TYPE_CHECKING:
@@ -133,9 +137,10 @@ class ProcMesh(MeshTrait):
         setup: Callable[[], None] | None = None,
     ) -> "ProcMesh":
         _rdma_manager = (
-            # pyre-ignore
+            # type: ignore[16]
             await _RdmaManager.create_rdma_manager_nonblocking(self._proc_mesh)
-            if HAS_TENSOR_ENGINE
+            # type: ignore[16]
+            if HAS_TENSOR_ENGINE and _RdmaBuffer.rdma_supported()
             else None
         )
 

monarch/tools/utils.py

@@ -53,13 +53,10 @@ class conda:
         Returns the currently active conda environment name.
         `None` if run outside of a conda environment.
         """
-        env_name = os.getenv("CONDA_DEFAULT_ENV")
-
-        if not env_name:
-            # conda envs activated with metaconda doesn't set CODNA_DEFAULT_ENV so
-            # fallback to CONDA_PREFIX which points to the path of the currently active conda environment
-            # e.g./home/$USER/.conda/envs/{env_name}
-            if env_dir := conda.active_env_dir():
-                env_name = os.path.basename(env_dir)
+        # we do not check CODNA_DEFAULT_ENV as CONDA_PREFIX is a preferred way
+        # to get the active conda environment, e.g./home/$USER/.conda/envs/{env_name}
+        env_name: Optional[str] = None
+        if env_dir := conda.active_env_dir():
+            env_name = os.path.basename(env_dir)
 
         return env_name

tests/test_coalescing.py

@@ -78,7 +78,7 @@ class TestCoalescing:
             num_hosts,
             gpu_per_host,
             activate,
-            backend=str(backend_type),
+            backend=backend_type.value,
         )
 
     @property

tests/test_python_actors.py

@@ -21,6 +21,7 @@ from typing import cast
 import pytest
 
 import torch
+from monarch._rust_bindings.monarch_hyperactor.pytokio import PythonTask
 
 from monarch._src.actor.actor_mesh import ActorMeshRef, Port, PortTuple
 
@@ -598,3 +599,16 @@ def test_ported_actor():
     proc_mesh = local_proc_mesh(gpus=1).get()
     a = proc_mesh.spawn("port_actor", PortedActor).get()
     assert 5 == a.add.call_one(2).get()
+
+
+async def _recv():
+    return (7, 2, 3)
+
+
+async def consume():
+    r = await PythonTask.from_coroutine(_recv())
+    assert r == (7, 2, 3)
+
+
+def test_python_task_tuple() -> None:
+    PythonTask.from_coroutine(consume()).block_on()

{torchmonarch_nightly-2025.7.31.dist-info → torchmonarch_nightly-2025.8.2.dist-info}/METADATA

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

{torchmonarch_nightly-2025.7.31.dist-info → torchmonarch_nightly-2025.8.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 monarch/__init__.py,sha256=mgKiyD1kxky-1pvhMlNfF4VmxWnhi-FSYZNFzkW1BEM,7052
-monarch/_rust_bindings.so,sha256=yXtU5ymNv_IeG2q-Eh8zVD-qmt7NGFFbp1XFbOPG2j0,50333048
+monarch/_rust_bindings.so,sha256=GpaO6AUerWPemfUCnIOMr2BrT2yG5MWyukPtgRGc_mc,50316552
 monarch/_testing.py,sha256=_3MYNMq-_0T1qXCj2vxrW13GlWGdUuVFMskQF2Gsw_o,7877
 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=JMxC8HdWMvpik0T4E6e-gfxvmNnOkA0ul4eo4R3Jg_o,1712
 monarch/gradient_generator.py,sha256=Rl3dmXGceTdCc1mYBg2JciR88ywGPnW7TVkL86KwqEA,6366
 monarch/memory.py,sha256=ol86dBhFAJqg78iF25-BuK0wuwj1onR8FIioZ_B0gjw,1377
 monarch/mesh_controller.py,sha256=R9ZnVV89wYva0QTAwOgHi_PkjYPEj_7_yF9810NHPak,14675
-monarch/monarch_controller,sha256=ZRuSJ2K5fz4cnCmD0jdoIpD5Xaj4o_HY9VNp-v2etqI,24026560
+monarch/monarch_controller,sha256=EGmkCLn670CnA8Pn-Wer9hj9s5daXEH20sD2VINWcWc,24031912
 monarch/notebook.py,sha256=zu9MKDFKf1-rCM2TqFSRJjMBeiWuKcJSyUFLvoZRQzs,25949
 monarch/opaque_module.py,sha256=jCcg0DjbcEVXA9WNG0NhUzGteLHOJLTZEBvrIYJIAns,10436
 monarch/opaque_object.py,sha256=x1LoX6RIMGh4ux52xIfhPgoh6PhZHdkf9bMccHW3DW0,2808
@@ -35,7 +35,7 @@ monarch/_src/actor/event_loop.py,sha256=2i4fKIkemBzua_t47BqVa2roZ6fWB6sbmMFPNx2z
 monarch/_src/actor/future.py,sha256=7QDiPu6-CnTw7cN_GWomQa9qGxDo5yXqCSqgyCJ7roU,5195
 monarch/_src/actor/pdb_wrapper.py,sha256=3pjk-eTSc7_rctDiZl-vilqTXQoaERGqyi1LueyoQGg,4342
 monarch/_src/actor/pickle.py,sha256=jD_3E07OJmMIlcMOOrNFnIuRKZU2F_Q_BP-njDFXUNM,2044
-monarch/_src/actor/proc_mesh.py,sha256=FNBRl-F1HdvdLWwHt0YrWYQhzXK4y1y-nSKnfwkznwA,15531
+monarch/_src/actor/proc_mesh.py,sha256=5AKgjsskVe7blJjwROom31p1EXzx9ANg3ADzvcLcU5s,15746
 monarch/_src/actor/shape.py,sha256=E9kxf1RNym1LNJNXF18gNDmnAHR7SDcl3W4nXR65BPY,8293
 monarch/_src/actor/sync_state.py,sha256=GB6bTAGztkcN8fZ9K7zXklOzjYzv6cvkJeBje20xFkE,471
 monarch/_src/actor/tensor_engine_shim.py,sha256=hupavQ2rjPwECaTlDAhY-eeiEY18Wyyx59MZHcSEcYM,1622
@@ -50,7 +50,7 @@ monarch/builtins/__init__.py,sha256=QcfnHZGbc2qktBg7DyZt2ruE6VahnIt4S8lEZLHdJqU,
 monarch/builtins/log.py,sha256=H1QkuVzwxyi36Zyv-XR0VN0QsNimBWwxE1__fjs0_2o,554
 monarch/builtins/random.py,sha256=wPbvscg7u53EXpMFo885fO2XOlsyjrNAJ4rBxLzfxdg,1839
 monarch/common/_C.pyi,sha256=kHY2G3ksMAjQJ6IcPb4F1bBh5knzw5RVVNhhBlEmwFU,314
-monarch/common/_C.so,sha256=i0Mqno03fQrMtkAnBwsenC4EzrU-NQy7fUk3-cGfOcw,719904
+monarch/common/_C.so,sha256=WkoR6BuS6k6tfia5sGSmfQVsBNmdKqRXXQ8o9pqfpy0,719808
 monarch/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monarch/common/_coalescing.py,sha256=HXf5cXAPSU_tpw9jFkzs2muytG_6sTZJSqSKV0XuFZE,10925
 monarch/common/_tensor_to_table.py,sha256=yRjCNwvtl188Z1Dwkx3ZU-Bh2mwYnQ0Lnue2RAztwvc,5753
@@ -89,7 +89,7 @@ monarch/controller/rust_backend/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTY
 monarch/controller/rust_backend/controller.py,sha256=8IYnVUiqEVKO9rGL3vKqcCSAhWJG1bYYQ0MoaMqsp78,9521
 monarch/gradient/__init__.py,sha256=kqmzwt16mMpk0M3GhpgP_f7da4DGnaV9chDzbt66k4Q,308
 monarch/gradient/_gradient_generator.pyi,sha256=6cX0UxaDt9NAlwgIhTgnweqGOf6qRhHiGnUzSWNCxdU,630
-monarch/gradient/_gradient_generator.so,sha256=Jw4MAcg1zcjYfm51N6EZWxTA8bedXaTcGMj3g45o9jQ,11534720
+monarch/gradient/_gradient_generator.so,sha256=bvJFYRlr0EKl4Qm4Qc0uHXMUop9IornqU5QqkCd_2C8,11756568
 monarch/parallel/__init__.py,sha256=6920kIkhiX7AiyjYvyc1ad8ccP-bStJJ1sS5KkeN2P0,352
 monarch/parallel/pipelining/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
 monarch/parallel/pipelining/runtime.py,sha256=KK8TG1gUYEzSsquiZoPTWGSIC74mlncD7cYknKxfb3c,32470
@@ -119,7 +119,7 @@ monarch/tools/cli.py,sha256=b3mKZnK-MwP7JwskTxHI0KcJXxSU6498jEb2ntVr_VM,5001
 monarch/tools/commands.py,sha256=3xuvHcMwl0t6cWTVUxI_r8EqrJZnay0bkKxOijhlKrw,12126
 monarch/tools/mesh_spec.py,sha256=in6txNRmA-UvveVSMHCjX6mGpofd3K8vl2Plz1eD6rg,7935
 monarch/tools/network.py,sha256=mN8Fx9mervxM3VdFHRn4ZXt4z7yWxZp52BTxx2tfpus,2455
-monarch/tools/utils.py,sha256=p9zunV_OP_eYC2GQFZ_NKFj5Xl1tf7Tin1ZxjZqp3TY,2010
+monarch/tools/utils.py,sha256=gcZyalfoBC6Y3v65h-QMngwXsn24ejXh2TH8RxlgXkA,1888
 monarch/tools/components/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
 monarch/tools/components/hyperactor.py,sha256=jbfC5J9oRzzMQFO2eIx9acRS8RPHp3GtKyJjYblcJFM,2169
 monarch/tools/config/__init__.py,sha256=HZjmRSC_R28WMDSvNAqNRlqhH7lMHFWPIrztcFFt8Us,890
@@ -148,7 +148,7 @@ tests/test_actor_error.py,sha256=U7QL1jRn-YpS-o62imt7HFLPtaSbwMBu9xpD09Mb-Bc,208
 tests/test_actor_shape.py,sha256=ph-RC9sMNHWptZOCwQqMfG4lIUEzhp_pEnfhITeYdHM,4533
 tests/test_alloc.py,sha256=IW7yJSaKxhOYc8SJtFyREakDUwiKWq9M0CGgYyBYHoc,743
 tests/test_allocator.py,sha256=DpQhygQ4jB19g-aY-BFR61J-gYbG-hDlTNPVuQ4Fmn0,29730
-tests/test_coalescing.py,sha256=JZ4YgQNlWWs7N-Z8KCCXQPANcuyyXEKjeHIXYbPnQhk,15606
+tests/test_coalescing.py,sha256=THQuYMoyb1idC7Y7b3B6YvhJolfjbRrBWG27senI7tE,15607
 tests/test_controller.py,sha256=CIMb-ApmBcBj1eCqccDUAbVyyJWMGooAha5gQk0AoeY,31452
 tests/test_debugger.py,sha256=9opgQXCBuZ1Z-7uOKI-FuGB0jLbLLilmWQKq0sE-dgQ,21950
 tests/test_device_mesh.py,sha256=DrbezYOM0thfP9MgLXb5-F0VoLOmSz5GR0GwjR_3bE4,5290
@@ -158,7 +158,7 @@ tests/test_future.py,sha256=cXzaNi2YDwVyjR541ScXmgktX1YFsKzbl8wep0DMVbk,3032
 tests/test_grad_generator.py,sha256=p4Pm4kMEeGldt2jUVAkGKCB0mLccKI28pltH6OTGbQA,3412
 tests/test_mock_cuda.py,sha256=5hisElxeLJ5MHw3KM9gwxBiXiMaG-Rm382u3AsQcDOI,3068
 tests/test_pdb_actor.py,sha256=5KJhuhcZDPWMdjC6eAtDdwnz1W7jNFXvIrMSFaCWaPw,3858
-tests/test_python_actors.py,sha256=kvsn2t2gmhnCM_T_euVe2O_DLj1OkzMrUNhhzBVPfs0,15664
+tests/test_python_actors.py,sha256=PA5nsIZa8V0RoT19jtwPKri7csYQQy3QgsshNLHJRUY,15969
 tests/test_rdma.py,sha256=vgeCCsfOjRjlGoGR0SYRuTP_Sx5RlEUUKfO9ATK0d4E,6125
 tests/test_remote_functions.py,sha256=VT65W7htp1jCsP9-AsiO1dofhx-4OebWEOssDEgi3GM,51054
 tests/test_rust_backend.py,sha256=3TLu8dSVEqyLhjHED2DoAEQHTpbBQcr3WI6K2eGZLZw,7861
@@ -170,9 +170,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.7.31.dist-info/licenses/LICENSE,sha256=e0Eotbf_rHOYPuEUlppIbvwy4SN98CZnl_hqwvbDA4Q,1530
-torchmonarch_nightly-2025.7.31.dist-info/METADATA,sha256=V4U9RkS-KxbvPEmkrhGYp3WRtTXluOVzAl9ktisydYw,3852
-torchmonarch_nightly-2025.7.31.dist-info/WHEEL,sha256=OlISbtpDcfagPrLwG7WtpcZbPTUnoKPnwphA_26fNqE,104
-torchmonarch_nightly-2025.7.31.dist-info/entry_points.txt,sha256=60QVSpYVzkzS4iDOiLp0fsLxVp47X3J2l3v7W-59LMo,117
-torchmonarch_nightly-2025.7.31.dist-info/top_level.txt,sha256=E-ZssZzyM17glpVrh-S9--qJ-w9p2EjuYOuNw9tQ4Eg,33
-torchmonarch_nightly-2025.7.31.dist-info/RECORD,,
+torchmonarch_nightly-2025.8.2.dist-info/licenses/LICENSE,sha256=e0Eotbf_rHOYPuEUlppIbvwy4SN98CZnl_hqwvbDA4Q,1530
+torchmonarch_nightly-2025.8.2.dist-info/METADATA,sha256=LJSxWhCMd2cXV4Kn_1ZXyEfHqPNP5QSoyTE-JXkzNek,3851
+torchmonarch_nightly-2025.8.2.dist-info/WHEEL,sha256=OlISbtpDcfagPrLwG7WtpcZbPTUnoKPnwphA_26fNqE,104
+torchmonarch_nightly-2025.8.2.dist-info/entry_points.txt,sha256=60QVSpYVzkzS4iDOiLp0fsLxVp47X3J2l3v7W-59LMo,117
+torchmonarch_nightly-2025.8.2.dist-info/top_level.txt,sha256=E-ZssZzyM17glpVrh-S9--qJ-w9p2EjuYOuNw9tQ4Eg,33
+torchmonarch_nightly-2025.8.2.dist-info/RECORD,,