Unbreak torch/distributed/rpc/batch (#1386)

msaroufim · web-flow · commit 7fce8bb0daf1 · 2025-08-25T13:01:33.000-07:00
* More RPC examples

* update
diff --git a/distributed/rpc/batch/parameter_server.py b/distributed/rpc/batch/parameter_server.py
@@ -129,7 +129,7 @@ def run(rank, world_size):
     
     options=rpc.TensorPipeRpcBackendOptions(
         num_worker_threads=16,
-        rpc_timeout=60  # 60 second timeout instead of infinite
+        rpc_timeout=60
      )
     if rank != 0:
         rpc.init_rpc(

Original file line number	Diff line number	Diff line change
`@@ -129,7 +129,7 @@ def run(rank, world_size):`
`129`	`129`
`130`	`130`	`options=rpc.TensorPipeRpcBackendOptions(`
`131`	`131`	`num_worker_threads=16,`
`132`		`- rpc_timeout=60 # 60 second timeout instead of infinite`
	`132`	`+ rpc_timeout=60`
`133`	`133`	`)`
`134`	`134`	`if rank != 0:`
`135`	`135`	`rpc.init_rpc(`