Merge pull request #13 from hpcaitech/feature/triton

make host and port the arguement
hpcaitech · Mar 18, 2022 · 45ee614 · 45ee614
2 parents e9c126f + ded230e
commit 45ee614
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/energon/engine/engine.py b/energon/engine/engine.py
@@ -50,7 +50,7 @@ def launch_rpc(tp_size: int = 1,
 
     processes = []
     for rank in range(world_size-1):
-        p = mp.Process(target=process_func, args=(tp_size, pp_size, backend, seed, verbose, rank+1, rank+1, world_size, 'localhost', 29500))
+        p = mp.Process(target=process_func, args=(tp_size, pp_size, backend, seed, verbose, rank+1, rank+1, world_size, host, port))
         p.start()
         processes.append(p)
 
@@ -139,4 +139,4 @@ def clear(self):
 
 
 
-
+