Updates all wheels and removes await (#272)

* update all wheels

* no awaits

* undo store/titan wheels

Author: allenwang28

assets/wheels/monarch-0.0.1-cp310-cp310-linux_x86_64.whl

@@ -17,7 +17,7 @@ NC='\033[0m'
 # Configuration
 PYTORCH_VERSION="2.9.0.dev20250905"
 VLLM_BRANCH="v0.10.0"
-MONARCH_COMMIT="ce078ddbbac6b6f3682afd763d8d96dad8092a91"
+MONARCH_COMMIT="9c41b5c16edadeab7cfb8521ba7efe68a1e2bc87"
 TORCHTITAN_COMMIT="a3104201ba3a0fa19e9c3cc5ba748b0398551410"
 TORCHSTORE_COMMIT="0052f6d8b686b9cff0cf4ce203a836c4b5d5ac94"
 BUILD_DIR="$HOME/forge-build"

scripts/build_wheels.sh

@@ -21,7 +21,7 @@ log_warning() { echo -e "${YELLOW}[WARNING]${NC} $1";}
 PYTORCH_VERSION="2.9.0.dev20250905"
 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 WHEEL_DIR="$SCRIPT_DIR/../assets/wheels"
-RELEASE_TAG="v0.0.0-92025"
+RELEASE_TAG="v0.0.0-93025"
 GITHUB_REPO="meta-pytorch/forge"
 
 # Check conda environment

scripts/install.sh

@@ -194,7 +194,7 @@ async def launch(  # pyright: ignore[reportIncompatibleMethodOverride]
         vllm_config = engine_config.create_vllm_config()
         # TODO (felipemello): LocalFetcherActor doesnt spawn with this, so cannot
         # do logging within PolicyWorker
-        workers = await worker_procs.spawn(
+        workers = worker_procs.spawn(
             "vllm_worker", PolicyWorker, vllm_config=vllm_config, use_dcp=use_dcp
         )
 
@@ -203,7 +203,7 @@ async def launch(  # pyright: ignore[reportIncompatibleMethodOverride]
 
         # TODO - expand support so name can stick within kwargs
         actor_name = kwargs.pop("name", cls.__name__)
-        policy = await policy_proc.spawn(
+        policy = policy_proc.spawn(
             actor_name,
             cls,
             engine_config=engine_config,

src/forge/actors/policy.py

@@ -6,21 +6,4 @@
 from .actor import ForgeActor
 from .proc_mesh import get_proc_mesh, stop_proc_mesh
 
-
-# TODO - remove this once everything has moved to
-# service
-async def spawn_actors(
-    name: str,
-    actor_cls: ForgeActor,
-    cfg,
-    processes,
-    set_address: bool = False,
-):
-    """Setup process Mesh and spawn Actors."""
-    mesh = await get_proc_mesh(processes)
-    actors = await mesh.spawn(name, actor_cls, **cfg)
-    actors.mesh = mesh
-    return actors
-
-
-__all__ = ["spawn_actors", "stop_proc_mesh", "get_proc_mesh", "ForgeActor"]
+__all__ = ["stop_proc_mesh", "get_proc_mesh", "ForgeActor"]

src/forge/controller/__init__.py

@@ -186,7 +186,7 @@ async def launch(cls, *args, **kwargs) -> "ForgeActor":
         proc_mesh = await get_proc_mesh(process_config=cfg)
 
         actor_name = kwargs.pop("name", cls.__name__)
-        actor = await proc_mesh.spawn(actor_name, cls, *args, **kwargs)
+        actor = proc_mesh.spawn(actor_name, cls, *args, **kwargs)
         actor._proc_mesh = proc_mesh
 
         if hasattr(proc_mesh, "_hostname") and hasattr(proc_mesh, "_port"):

src/forge/controller/actor.py

@@ -199,7 +199,7 @@ def bootstrap(gpu_ids: list[str]):
                     per_host={"gpus": num_procs},
                     bootstrap=functools.partial(bootstrap, gpu_ids=gpu_ids),
                 )
-                setup = await procs.spawn(f"setup-{uuid.uuid1()}", _SetupActor)
+                setup = procs.spawn(f"setup-{uuid.uuid1()}", _SetupActor)
                 # Pick a random host/port, we'll feed this in afterwards
                 # Once we have true HostMesh support, we can do this on proc 0 of each host
                 # then spin up the proc meshes with the environment afterwards.

src/forge/controller/provisioner.py

@@ -41,7 +41,7 @@ async def spawn_service_v2(
     # Create a single-node proc_mesh and actor_mesh for the Service Actor
     logger.info("Spawning Service Actor for %s", actor_def.__name__)
     m = await proc_mesh(gpus=1)
-    service_actor = await m.spawn(
+    service_actor = m.spawn(
         "service", ServiceActor, service_cfg, actor_def, actor_kwargs
     )
     await service_actor.__initialize__.call_one()

src/forge/controller/service/spawn.py

@@ -97,7 +97,7 @@ async def get_or_create_metric_logger(
 
     # Setup local_fetcher_actor if needed
     if not proc_has_local_fetcher:
-        local_fetcher_actor = await proc.spawn(
+        local_fetcher_actor = proc.spawn(
             "local_fetcher_actor", LocalFetcherActor, global_logger
         )
         await global_logger.register_fetcher.call_one(local_fetcher_actor, proc)