oneflow.env.init_rdma¶
-
oneflow.env.
init_rdma
()¶ Init RDMA in the current envirment. If the current envirment support RDMA, turning on RDMA by calling oneflow.env.init_rdma() can speed up data transfer.
Note
Make sure to avoid using fork() after oneflow.env.init_rdma() is invoked. Otherwise, data corruption or segmentation fault may result!
Requires all devices to execute oneflow.env.init_rdma() simultaneously. Otherwise, deadlock may result!