We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4db1116 commit 9f9cd44Copy full SHA for 9f9cd44
distributed/rpc/pipeline/main.py
@@ -219,7 +219,9 @@ def run_master(split_size):
219
def run_worker(rank, world_size, num_split):
220
os.environ['MASTER_ADDR'] = 'localhost'
221
os.environ['MASTER_PORT'] = '29500'
222
- options = rpc.TensorPipeRpcBackendOptions(num_worker_threads=256)
+
223
+ # Higher timeout is added to accommodate for kernel compilation time in case of ROCm.
224
+ options = rpc.TensorPipeRpcBackendOptions(num_worker_threads=256, rpc_timeout=300)
225
226
if rank == 0:
227
rpc.init_rpc(
0 commit comments