dmlc--dgl
Minjie Wang
025e43210f
* pass gcn * example tests * pass gcnii, hgnn, sgc, sign * black * test file * add torchmetrics * migrate appnp, c_and_s, twirls * fix style * minor fix * fix hypergraphatt and twirls * pass gat * fix torchmetric * lintrunner Co-authored-by: Mufei Li <mufeili1996@gmail.com>
44 行
1.1 KiB
Bash
44 行
1.1 KiB
Bash
#!/bin/bash
|
|
|
|
. /opt/conda/etc/profile.d/conda.sh
|
|
|
|
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}
|
|
export TF_FORCE_GPU_ALLOW_GROWTH=true
|
|
unset TORCH_ALLOW_TF32_CUBLAS_OVERRIDE
|
|
|
|
if [ $2 == "gpu" ]
|
|
then
|
|
export CUDA_VISIBLE_DEVICES=0
|
|
else
|
|
export CUDA_VISIBLE_DEVICES=-1
|
|
fi
|
|
|
|
conda activate ${DGLBACKEND}-ci
|
|
|
|
python3 -m pip install pytest psutil pyyaml pydantic pandas rdflib ogb torchmetrics || fail "pip install"
|
|
if [ $DGLBACKEND == "mxnet" ]
|
|
then
|
|
python3 -m pytest -v --junitxml=pytest_compute.xml --durations=100 --ignore=tests/compute/test_ffi.py tests/compute || fail "compute"
|
|
else
|
|
python3 -m pytest -v --junitxml=pytest_compute.xml --durations=100 tests/compute || fail "compute"
|
|
fi
|
|
python3 -m pytest -v --junitxml=pytest_backend.xml --durations=100 tests/$DGLBACKEND || fail "backend-specific"
|