Disable ibv in rpc

alibaba · Nov 6, 2023 · 3138546 · 3138546
1 parent 7b9e175
commit 3138546
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/graphlearn_torch/python/distributed/rpc.py b/graphlearn_torch/python/distributed/rpc.py
@@ -255,7 +255,8 @@ def init_rpc(master_addr: str,
       raise RuntimeError("'init_rpc': Distributed context has not been set.")
 
     options = rpc.TensorPipeRpcBackendOptions(
-      _transports=['ibv', 'uv'],
+      # _transports=['ibv', 'uv'],
+      _transports=['uv'],
       _channels=['mpt_uv', 'basic'],
       num_worker_threads=num_rpc_threads,
       rpc_timeout=rpc_timeout,