"vscode:/vscode.git/clone" did not exist on "d3483fe1f11a73b19253b975221a48717d288edf"
task_dist_test.sh 1.09 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/bin/bash
function fail {
    echo FAIL: $@
    exit -1
}

echo $PWD
export DGLBACKEND=pytorch
export DGL_LIBRARY_PATH=${PWD}/build
export PYTHONPATH=${PWD}/tests:${PWD}/python:$PYTHONPATH
export LD_LIBRARY_PATH=${PWD}/build:$LD_LIBRARY_PATH
export DIST_DGL_TEST_CPP_BIN_DIR=${PWD}/build
export DIST_DGL_TEST_IP_CONFIG=/home/ubuntu/workspace/ip_config.txt
export DIST_DGL_TEST_PY_BIN_DIR=${PWD}/tests/dist/python

16
17
18
19
20
21
22
23
24
25
26
27
28
if [[ -v DIST_DGL_TEST_SSH_PORT ]]; then
    SSH_PORT_LINE="-p $DIST_DGL_TEST_SSH_PORT";
fi

if [[ -v DIST_DGL_TEST_SSH_KEY ]]; then
    SSH_KEY_LINE="-i $DIST_DGL_TEST_SSH_KEY";
fi

if [[ -v DIST_DGL_TEST_SSH_SETUP ]]; then
    SSH_SETUP_LINE="$DIST_DGL_TEST_SSH_SETUP;";
fi


29
30
31
32
while IFS= read line
do
    for pkg in 'pytest' 'psutil' 'torch'
    do
33
        ret_pkg=$(ssh -o StrictHostKeyChecking=no ${line} ${SSH_PORT_LINE} ${SSH_KEY_LINE} "${SSH_SETUP_LINE}python3 -m pip list | grep -i ${pkg} ") || fail "${pkg} not installed in ${line}"
34
35
36
37
    done
done < ${DIST_DGL_TEST_IP_CONFIG}

python3 -m pytest -v --capture=tee-sys --junitxml=pytest_dist.xml tests/dist/test_*.py || fail "dist across machines"