torchmonarch-nightly 2025.7.28__cp311-cp311-manylinux2014_x86_64.whl → 2025.7.29__cp311-cp311-manylinux2014_x86_64.whl

monarch/_src/actor/allocator.py

@@ -8,7 +8,7 @@
 
 import abc
 import logging
-from typing import final, Optional
+from typing import Awaitable, final, Optional, TYPE_CHECKING
 
 from monarch._rust_bindings.monarch_hyperactor.alloc import (  # @manual=//monarch/monarch_extension:monarch_extension
     Alloc,
@@ -19,6 +19,9 @@ from monarch._rust_bindings.monarch_hyperactor.alloc import (  # @manual=//monar
     SimAllocatorBase,
 )
 
+if TYPE_CHECKING:
+    from monarch._rust_bindings.monarch_hyperactor.tokio import PythonTask
+
 from monarch._src.actor.future import Future
 
 ALLOC_LABEL_PROC_MESH_NAME = "procmesh.monarch.meta.com/name"
@@ -26,11 +29,9 @@ ALLOC_LABEL_PROC_MESH_NAME = "procmesh.monarch.meta.com/name"
 logger: logging.Logger = logging.getLogger(__name__)
 
 
-@final
-class ProcessAllocator(ProcessAllocatorBase):
-    """
-    An allocator that allocates by spawning local processes.
-    """
+class AllocateMixin(abc.ABC):
+    @abc.abstractmethod
+    def allocate_nonblocking(self, spec: AllocSpec) -> "Awaitable[Alloc]": ...
 
     def allocate(self, spec: AllocSpec) -> Future[Alloc]:
         """
@@ -46,42 +47,25 @@ class ProcessAllocator(ProcessAllocatorBase):
 
 
 @final
-class LocalAllocator(LocalAllocatorBase):
+class ProcessAllocator(ProcessAllocatorBase, AllocateMixin):
     """
-    An allocator that allocates by spawning actors into the current process.
+    An allocator that allocates by spawning local processes.
     """
 
-    def allocate(self, spec: AllocSpec) -> Future[Alloc]:
-        """
-        Allocate a process according to the provided spec.
 
-        Arguments:
-        - `spec`: The spec to allocate according to.
-
-        Returns:
-        - A future that will be fulfilled when the requested allocation is fulfilled.
-        """
-        return Future(impl=lambda: self.allocate_nonblocking(spec), requires_loop=False)
+@final
+class LocalAllocator(LocalAllocatorBase, AllocateMixin):
+    """
+    An allocator that allocates by spawning actors into the current process.
+    """
 
 
 @final
-class SimAllocator(SimAllocatorBase):
+class SimAllocator(SimAllocatorBase, AllocateMixin):
     """
     An allocator that allocates by spawning actors into the current process using simulated channels for transport
     """
 
-    def allocate(self, spec: AllocSpec) -> Future[Alloc]:
-        """
-        Allocate a process according to the provided spec.
-
-        Arguments:
-        - `spec`: The spec to allocate according to.
-
-        Returns:
-        - A future that will be fulfilled when the requested allocation is fulfilled.
-        """
-        return Future(impl=lambda: self.allocate_nonblocking(spec), requires_loop=False)
-
 
 class RemoteAllocInitializer(abc.ABC):
     """Subclass-able Python interface for `hyperactor_mesh::alloc::remoteprocess:RemoteProcessAllocInitializer`.
@@ -210,20 +194,8 @@ class TorchXRemoteAllocInitializer(RemoteAllocInitializer):
 
 
 @final
-class RemoteAllocator(RemoteAllocatorBase):
+class RemoteAllocator(RemoteAllocatorBase, AllocateMixin):
     """
     An allocator that allocates by spawning actors on a remote host.
     The remote host must be running hyperactor's remote-process-allocator.
     """
-
-    def allocate(self, spec: AllocSpec) -> Future[Alloc]:
-        """
-        Allocate a process according to the provided spec.
-
-        Arguments:
-        - `spec`: The spec to allocate according to.
-
-        Returns:
-        - A future that will be fulfilled when the requested allocation is fulfilled.
-        """
-        return Future(impl=lambda: self.allocate_nonblocking(spec), requires_loop=False)

monarch/_src/actor/shape.py

@@ -67,7 +67,12 @@ class MeshTrait(ABC):
                     start, stop, slice_stride = e.indices(size)
                     offset += start * stride
                     names.append(name)
-                    sizes.append((stop - start) // slice_stride)
+                    # The number of elems in `start..stop` with step
+                    # `slice_stride`. This is:
+                    #    ⌈(stop - start) /slice_stride⌉
+                    # — the number of stride steps that fit in the
+                    # half-open interval.
+                    sizes.append((stop - start + slice_stride - 1) // slice_stride)
                     strides.append(slice_stride * stride)
                 else:
                     if e >= size or e < 0:

monarch/tools/components/hyperactor.py

@@ -22,7 +22,7 @@ DEFAULT_NAME: str = f"monarch-{_USER}"
 __version__ = "latest"  # TODO get version from monarch.__version_
 
 
-def proc_mesh(
+def host_mesh(
     image: str = f"ghcr.io/pytorch-labs/monarch:{__version__}",  # TODO docker needs to be built and pushed to ghcr
     meshes: list[str] = _DEFAULT_MESHES,
     env: Optional[dict[str, str]] = None,

monarch/tools/config/__init__.py

@@ -34,4 +34,4 @@ class Config:
     scheduler_args: dict[str, Any] = field(default_factory=dict)
     workspace: Optional[str] = None
     dryrun: bool = False
-    appdef: UnnamedAppDef = UnnamedAppDef()
+    appdef: UnnamedAppDef = field(default_factory=UnnamedAppDef)

monarch/tools/config/defaults.py

@@ -25,7 +25,7 @@ from torchx.schedulers import (
 
 def component_fn(scheduler: str) -> Callable[..., UnnamedAppDef]:
     """The default TorchX component function for the scheduler"""
-    return hyperactor.proc_mesh
+    return hyperactor.host_mesh
 
 
 def scheduler_factories() -> dict[str, SchedulerFactory]:

monarch/tools/utils.py

@@ -6,9 +6,36 @@
 
 # pyre-strict
 import os
+import pathlib
 from typing import Optional
 
 
+def MONARCH_HOME(*subdir_paths: str) -> pathlib.Path:
+    """
+    Path to the "dot-directory" for monarch.
+    Defaults to `~/.monarch` and is overridable via the `MONARCH_HOME` environment variable.
+
+    Usage:
+
+    .. doc-test::
+
+        from pathlib import Path
+        from monarch.tools.utils import MONARCH_HOME
+
+        assert MONARCH_HOME() == Path.home() / ".monarch"
+        assert MONARCH_HOME("conda-pack-out") ==  Path.home() / ".monarch" / "conda-pack-out"
+    ```
+    """
+
+    default_dir = str(pathlib.Path.home() / ".monarch")
+    monarch_home = pathlib.Path(os.getenv("MONARCH_HOME", default_dir))
+
+    monarch_home_subdir = monarch_home / os.path.sep.join(subdir_paths)
+    monarch_home_subdir.mkdir(parents=True, exist_ok=True)
+
+    return monarch_home_subdir
+
+
 class conda:
     """Conda related util functions."""
 

{torchmonarch_nightly-2025.7.28.dist-info → torchmonarch_nightly-2025.7.29.dist-info}/METADATA

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

{torchmonarch_nightly-2025.7.28.dist-info → torchmonarch_nightly-2025.7.29.dist-info}/RECORD

@@ -1,5 +1,5 @@
 monarch/__init__.py,sha256=mgKiyD1kxky-1pvhMlNfF4VmxWnhi-FSYZNFzkW1BEM,7052
-monarch/_rust_bindings.so,sha256=crjfqvnpPBQUV-9OeCrlZ0WBq8xEcJyX8mvK1yWsp-8,50483080
+monarch/_rust_bindings.so,sha256=ScSFD_QzNVyFbKzJ7o315q8R71cj-LtiHXB2PEqwYFM,50485024
 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=mOk2misobJun2AgR_ALjFoopAEcOPYQVrrAJXa18ZTs,13810
-monarch/monarch_controller,sha256=untvTxNKGEpzfX3Eh2InHETFphEaRgzxwWPwHWdroOg,24053432
+monarch/monarch_controller,sha256=Sgt5YAW4P6-y8UOTShKyGvpMlUu2EBKif6lfAJE6k3s,24055184
 monarch/notebook.py,sha256=zu9MKDFKf1-rCM2TqFSRJjMBeiWuKcJSyUFLvoZRQzs,25949
 monarch/opaque_module.py,sha256=jCcg0DjbcEVXA9WNG0NhUzGteLHOJLTZEBvrIYJIAns,10436
 monarch/opaque_object.py,sha256=x1LoX6RIMGh4ux52xIfhPgoh6PhZHdkf9bMccHW3DW0,2808
@@ -26,7 +26,7 @@ monarch/world_mesh.py,sha256=ob5dJWaC49Uw0xqClHBm8CQLvL4xKnjd4TGzk7k8NxI,980
 monarch/_src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monarch/_src/actor/__init__.py,sha256=4iK3nzQZmEPe0HGNY70fABBenw3lCVVaaF0xddF5Fa0,235
 monarch/_src/actor/actor_mesh.py,sha256=wmFUf3NJcHmJeK7WrLkGOrzmAvdHmjLUaeQ5W6WDLk0,30867
-monarch/_src/actor/allocator.py,sha256=WpHEK1SvjgF3GdIWIYUkonXli2-gQVKJVZPInl2RFQo,8212
+monarch/_src/actor/allocator.py,sha256=KUmZYJdNK_k0R_59cXf1EntmD2K5fbptWXBA3-4Txvw,7303
 monarch/_src/actor/bootstrap_main.py,sha256=e5eU3GvX60MWWmCty7VcZrAmukD29iJdWBysNgQ2o3A,2342
 monarch/_src/actor/debugger.py,sha256=t2iAAxz03b2KZ89T3VjRc__7GzSf83R8gM81SDyX3-o,19532
 monarch/_src/actor/device_utils.py,sha256=gBpl23wMjppVAEzzj8U9HyX-B7Bs2_3ftiMAkzUS4j4,577
@@ -36,7 +36,7 @@ monarch/_src/actor/future.py,sha256=jOGh1wfwKyGJxhl9t1P8eapXYywf8KwQldZCCbupmb8,
 monarch/_src/actor/pdb_wrapper.py,sha256=-QxRktntdEO2LdHixBGKLboYtADyh8bEIAoa3gFwIEo,4161
 monarch/_src/actor/pickle.py,sha256=jD_3E07OJmMIlcMOOrNFnIuRKZU2F_Q_BP-njDFXUNM,2044
 monarch/_src/actor/proc_mesh.py,sha256=amF4fbO-33qHFudlS9WabYXIVh0Y_D_0nhCTxvOhpGg,16640
-monarch/_src/actor/shape.py,sha256=B-7DI768ZhT8ECUNCJcI7DfCB7iDFGFH0r-HmXaAfcM,8296
+monarch/_src/actor/shape.py,sha256=_vbZW36KO4MN3MDBp2B7B1RkYQXRylj7HHoYS4SJx50,8602
 monarch/_src/actor/sync_state.py,sha256=GB6bTAGztkcN8fZ9K7zXklOzjYzv6cvkJeBje20xFkE,471
 monarch/_src/actor/tensor_engine_shim.py,sha256=hupavQ2rjPwECaTlDAhY-eeiEY18Wyyx59MZHcSEcYM,1622
 monarch/_src/actor/code_sync/__init__.py,sha256=qzWoFNJEJvEbqab0QuHbkvhdz6FHi7BOTw6-2B3p0A4,378
@@ -119,11 +119,11 @@ 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=2GGUQQE0dLtzoKy40_tAsOfbSxE6krnL0WvwMgUBgmw,1213
+monarch/tools/utils.py,sha256=p9zunV_OP_eYC2GQFZ_NKFj5Xl1tf7Tin1ZxjZqp3TY,2010
 monarch/tools/components/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
-monarch/tools/components/hyperactor.py,sha256=gYZS8AcmoTuq48mRrZWWnyxQqaiwTNHv8YqHhHi799U,2169
-monarch/tools/config/__init__.py,sha256=MLa6uvVJssN_zTciCvCMeCURWglchCuqE3zdqA-gh4U,869
-monarch/tools/config/defaults.py,sha256=ZymyKr9fNnBIgsV-xz-KrtrXRLkJo3hymTqxjXXnBzs,1910
+monarch/tools/components/hyperactor.py,sha256=jbfC5J9oRzzMQFO2eIx9acRS8RPHp3GtKyJjYblcJFM,2169
+monarch/tools/config/__init__.py,sha256=HZjmRSC_R28WMDSvNAqNRlqhH7lMHFWPIrztcFFt8Us,890
+monarch/tools/config/defaults.py,sha256=7YxdL8JHdjzlI2hTCgyYVgIbViJW8gfwlpcXvgfaKh0,1910
 monarch/worker/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
 monarch/worker/_testing_function.py,sha256=A81cVMKgdlO66XvoYcBCDrxIQIm3o3GgvcH_c8M9OmI,13480
 monarch/worker/compiled_block.py,sha256=hYx1F6PAu0_BnpKAprP_nV9qJtk5XWO7mcwH3JPDioU,10114
@@ -169,9 +169,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.28.dist-info/licenses/LICENSE,sha256=e0Eotbf_rHOYPuEUlppIbvwy4SN98CZnl_hqwvbDA4Q,1530
-torchmonarch_nightly-2025.7.28.dist-info/METADATA,sha256=empMLiZrrLDR6taSJTViJ0CSOI3B1Jc6kI3H6wDNd8w,3852
-torchmonarch_nightly-2025.7.28.dist-info/WHEEL,sha256=JC9FVdjbTDi9l3EyrqUd11CgmN9LkBi1g5dFHayafwA,104
-torchmonarch_nightly-2025.7.28.dist-info/entry_points.txt,sha256=60QVSpYVzkzS4iDOiLp0fsLxVp47X3J2l3v7W-59LMo,117
-torchmonarch_nightly-2025.7.28.dist-info/top_level.txt,sha256=E-ZssZzyM17glpVrh-S9--qJ-w9p2EjuYOuNw9tQ4Eg,33
-torchmonarch_nightly-2025.7.28.dist-info/RECORD,,
+torchmonarch_nightly-2025.7.29.dist-info/licenses/LICENSE,sha256=e0Eotbf_rHOYPuEUlppIbvwy4SN98CZnl_hqwvbDA4Q,1530
+torchmonarch_nightly-2025.7.29.dist-info/METADATA,sha256=MZ_yZwBZESYMwztGOrfuxFXmKgiojuezNPNHptuMcPc,3852
+torchmonarch_nightly-2025.7.29.dist-info/WHEEL,sha256=JC9FVdjbTDi9l3EyrqUd11CgmN9LkBi1g5dFHayafwA,104
+torchmonarch_nightly-2025.7.29.dist-info/entry_points.txt,sha256=60QVSpYVzkzS4iDOiLp0fsLxVp47X3J2l3v7W-59LMo,117
+torchmonarch_nightly-2025.7.29.dist-info/top_level.txt,sha256=E-ZssZzyM17glpVrh-S9--qJ-w9p2EjuYOuNw9tQ4Eg,33
+torchmonarch_nightly-2025.7.29.dist-info/RECORD,,