dmlc--dgl
Da Zheng
2e9949d228
* fix graph store for Pytorch. * add test. * fix dtype error in test * disable test on GPU. * test avoid windows. * fix shared-memory test. * use script to control testing environment. * update test. * enable all tests. * fix test script.
30 行
707 B
Bash
30 行
707 B
Bash
#!/bin/bash
|
|
|
|
function fail {
|
|
echo FAIL: $@
|
|
exit -1
|
|
}
|
|
|
|
function usage {
|
|
echo "Usage: $0 backend device"
|
|
}
|
|
|
|
if [ $# -ne 2 ]; then
|
|
usage
|
|
fail "Error: must specify backend and device"
|
|
fi
|
|
|
|
export DGLBACKEND=$1
|
|
export DGLTESTDEV=$2
|
|
export DGL_LIBRARY_PATH=${PWD}/build
|
|
export PYTHONPATH=tests:${PWD}/python:$PYTHONPATH
|
|
export DGL_DOWNLOAD_DIR=${PWD}
|
|
|
|
python3 -m nose -v --with-xunit tests/compute || fail "compute"
|
|
python3 -m nose -v --with-xunit tests/graph_index || fail "graph_index"
|
|
python3 -m nose -v --with-xunit tests/$DGLBACKEND || fail "backend-specific"
|
|
export OMP_NUM_THREADS=1
|
|
if [ $2 != "gpu" ]; then
|
|
python3 -m nose -v --with-xunit tests/distributed || fail "distributed"
|
|
fi
|