Change test to use tensorpipe rpc backend (#759)

57821dd2 · Howard Huang · GitHub · 3e661603 · 57821dd2
Unverified Commit 57821dd2 authored Aug 02, 2021 by Howard Huang Committed by GitHub Aug 02, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

benchmarks/experimental/experimental_async_approaches.py benchmarks/experimental/experimental_async_approaches.py +2 -2

No files found.
--- a/benchmarks/experimental/experimental_async_approaches.py
+++ b/benchmarks/experimental/experimental_async_approaches.py
@@ -815,8 +815,8 @@ def bench_mpi(args):
        f"Test{rank}",
        rank=rank,
        world_size=world_size,
-        backend=rpc.BackendType.PROCESS_GROUP,
+        backend=rpc.BackendType.TENSORPIPE,
-        rpc_backend_options=rpc.ProcessGroupRpcBackendOptions(rpc_timeout=20, init_method=init_method),
+        rpc_backend_options=rpc.TensorPipeRpcBackendOptions(rpc_timeout=20, init_method=init_method),
    )
    backends = {"model_parallel_backend": "nccl", "pipeline_backend": "mpi", "ddp_backend": "nccl"}