Skip to content

Commit 4f15ad4

Browse files
HosseinKaviani-HHossein Kavianihamedaniallenwang28felipemello1
authored andcommitted
Fix SLURM Launcher: Add Hyperactor Import and Use TcpWithHostname Transport (meta-pytorch#582)
Co-authored-by: Hossein Kavianihamedani <[email protected]> Co-authored-by: Allen Wang <[email protected]> Co-authored-by: Felipe Mello <[email protected]>
1 parent 9a9b79f commit 4f15ad4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/forge/controller/launcher.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
from monarch.actor import Actor, endpoint, ProcMesh
2828
from monarch.tools import commands
2929
from monarch.tools.commands import create, info
30+
from monarch.tools.components import hyperactor
3031
from monarch.tools.config import Config, Workspace
3132

3233
_MAST_AVAILABLE = False
@@ -128,7 +129,7 @@ async def initialize(self) -> None:
128129
# HostMesh currently requires explicit configuration
129130
# of the underlying transport from client to mesh.
130131
# This can be removed in the future once this has been removed.
131-
configure(default_transport=ChannelTransport.Tcp)
132+
configure(default_transport=ChannelTransport.TcpWithHostname)
132133

133134
async def get_allocator(self, name: str, num_hosts: int) -> tuple[Any, Any, str]:
134135
appdef = hyperactor.host_mesh(

0 commit comments

Comments
 (0)