dmlc--dgl
Rhett Ying
c1ff4c9b41
* [DistTest] add basic pipeline for dist test across machines * move launch remote cmd to separate file * add test for rpc * fix function naming rule Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
23 行
775 B
Python
23 行
775 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
|