dmlc--dgl
Hongzhi (Steve), Chen
89a4cc4db2
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
25 行
778 B
Python
25 行
778 B
Python
import os
|
|
import unittest
|
|
|
|
from utils import execute_remote, get_ips
|
|
|
|
|
|
@unittest.skipIf(os.name == "nt", reason="Do not support windows yet")
|
|
def test_tensorpipe_comm():
|
|
base_dir = os.environ.get("DIST_DGL_TEST_CPP_BIN_DIR", ".")
|
|
ip_config = os.environ.get("DIST_DGL_TEST_IP_CONFIG", "ip_config.txt")
|
|
client_bin = os.path.join(base_dir, "rpc_client")
|
|
server_bin = os.path.join(base_dir, "rpc_server")
|
|
ips = get_ips(ip_config)
|
|
num_machines = len(ips)
|
|
procs = []
|
|
for ip in ips:
|
|
procs.append(
|
|
execute_remote(server_bin + " " + str(num_machines) + " " + ip, ip)
|
|
)
|
|
for ip in ips:
|
|
procs.append(execute_remote(client_bin + " " + ip_config, ip))
|
|
for p in procs:
|
|
p.join()
|
|
assert p.exitcode == 0
|