mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-05 19:54:25 +08:00
f7dae4533a
* try optimize CI * fix go test; adjust timing report * disable certain tests for mx/tf backends * fix ut * add pydantic
38 lines
1.1 KiB
Bash
38 lines
1.1 KiB
Bash
#!/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
|
|
|
|
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
|
|
|
|
|
|
while IFS= read line
|
|
do
|
|
for pkg in 'pytest' 'psutil' 'torch'
|
|
do
|
|
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}"
|
|
done
|
|
done < ${DIST_DGL_TEST_IP_CONFIG}
|
|
|
|
python3 -m pytest -v --capture=tee-sys --junitxml=pytest_dist.xml --durations=100 tests/dist/test_*.py || fail "dist across machines"
|