dmlc--dgl
Mufei Li
2e8ae9f980
* chunked graph data format * Update * Update * Update task_distributed_test.sh * Update * Update * Revert "Update" This reverts commit 03c461870f19375fb03125b061fc853ab555577f. * Update * Update * ssh-keygen * CI * install openssh * openssh * Update * CI * Update * Update Co-authored-by: Ubuntu <ubuntu@ip-172-31-53-142.us-west-2.compute.internal> Co-authored-by: Ubuntu <ubuntu@ip-172-31-16-87.us-west-2.compute.internal> Co-authored-by: Ubuntu <ubuntu@ip-172-31-20-21.us-west-2.compute.internal> Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-26.ap-northeast-1.compute.internal>
42 行
1.1 KiB
Bash
42 行
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
|
|
|
|
[ $1 == "pytorch" ] || fail "Distrbuted tests run on pytorch backend only."
|
|
[ $2 == "cpu" ] || fail "Distrbuted tests run on cpu only."
|
|
|
|
export DGLBACKEND=$1
|
|
export DGLTESTDEV=$2
|
|
export DGL_LIBRARY_PATH=${PWD}/build
|
|
export PYTHONPATH=tests:${PWD}/python:$PYTHONPATH
|
|
export DGL_DOWNLOAD_DIR=${PWD}
|
|
unset TORCH_ALLOW_TF32_CUBLAS_OVERRIDE
|
|
|
|
export CUDA_VISIBLE_DEVICES=-1
|
|
|
|
conda activate ${DGLBACKEND}-ci
|
|
|
|
python3 -m pip install pytest psutil pyyaml pydantic pandas rdflib ogb filelock pyarrow || fail "pip install"
|
|
|
|
export PYTHONUNBUFFERED=1
|
|
export OMP_NUM_THREADS=1
|
|
export DMLC_LOG_DEBUG=1
|
|
|
|
python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml --durations=100 tests/distributed/*.py || fail "distributed"
|
|
|
|
PYTHONPATH=tools:$PYTHONPATH python3 -m pytest -v --capture=tee-sys --junitxml=pytest_tools.xml --durations=100 tests/tools/*.py || fail "tools"
|