dmlc--dgl
Rhett Ying
85f281170f
* [CI] add new stage specific forcuda related features based on nvidia+pytorch * build and test for gpu_nv * fix build failure * fix unit tests * make -j * install cython beforehand * copy cython lib * test cugraph tests only * fix typo * separate test script for cugraph * refactor build dgl shell
22 行
522 B
Bash
22 行
522 B
Bash
#!/bin/bash
|
|
|
|
. /opt/conda/etc/profile.d/conda.sh
|
|
|
|
function fail {
|
|
echo FAIL: $@
|
|
exit -1
|
|
}
|
|
|
|
export DGLBACKEND=$1
|
|
export DGLTESTDEV=gpu
|
|
export DGL_LIBRARY_PATH=${PWD}/build
|
|
export PYTHONPATH=tests:${PWD}/python:$PYTHONPATH
|
|
export DGL_DOWNLOAD_DIR=${PWD}
|
|
export TF_FORCE_GPU_ALLOW_GROWTH=true
|
|
|
|
export CUDA_VISIBLE_DEVICES=0
|
|
|
|
python3 -m pip install pytest psutil pyyaml pydantic pandas rdflib ogb || fail "pip install"
|
|
|
|
python3 -m pytest -v --junitxml=pytest_cugraph.xml --durations=20 tests/cugraph || fail "cugraph"
|