Skip to content

Commit a4be7e6

Browse files
author
Hossein Kavianihamedani
committed
Fix launcher issues
1 parent 6e77f0b commit a4be7e6

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/forge/controller/launcher.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
from monarch.actor import Actor, endpoint, ProcMesh
2626
from monarch.tools import commands
2727
from monarch.tools.commands import create, info
28+
from monarch.tools.components import hyperactor
2829
from monarch.tools.config import Config, Workspace
2930

3031
_MAST_AVAILABLE = False
@@ -126,7 +127,8 @@ async def initialize(self) -> None:
126127
# HostMesh currently requires explicit configuration
127128
# of the underlying transport from client to mesh.
128129
# This can be removed in the future once this has been removed.
129-
configure(default_transport=ChannelTransport.Tcp)
130+
# Changed TCP this to TcpWithHostname for the slurm launcher
131+
configure(default_transport=ChannelTransport.TcpWithHostname)
130132

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

0 commit comments

Comments
 (0)