Skip to content

GitLab

Commit ea0899d0 authored by lisj's avatar lisj
Browse files

修复测试脚本的环境设置

parent 614007d7
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/scripts/task_unit_test.sh
View file @ ea0899d0
...@@ -22,6 +22,7 @@ export DGL_LIBRARY_PATH=${PWD}/build ...@@ -22,6 +22,7 @@ export DGL_LIBRARY_PATH=${PWD}/build
export PYTHONPATH=tests:${PWD}/python:$PYTHONPATH export PYTHONPATH=tests:${PWD}/python:$PYTHONPATH
export DGL_DOWNLOAD_DIR=${PWD} export DGL_DOWNLOAD_DIR=${PWD}
export TF_FORCE_GPU_ALLOW_GROWTH=true export TF_FORCE_GPU_ALLOW_GROWTH=true
export LD_LIBRARY_PATH=/usr/local/lib/python3.8/site-packages/torch/lib/:$LD_LIBRARY_PATH
unset TORCH_ALLOW_TF32_CUBLAS_OVERRIDE unset TORCH_ALLOW_TF32_CUBLAS_OVERRIDE
if [ $2 == "gpu" ] if [ $2 == "gpu" ]
...@@ -33,7 +34,7 @@ fi ...@@ -33,7 +34,7 @@ fi
conda activate ${DGLBACKEND}-ci conda activate ${DGLBACKEND}-ci
python3 -m pip install pytest psutil pyyaml pydantic pandas rdflib ogb || fail "pip install" python3 -m pip install pytest psutil pyyaml pydantic pandas rdflib ogb urllib3==1.26.15 || fail "pip install"
if [ $DGLBACKEND == "mxnet" ] if [ $DGLBACKEND == "mxnet" ]
then then
python3 -m pytest -v --junitxml=pytest_compute.xml --durations=100 --ignore=tests/compute/test_ffi.py tests/compute || fail "compute" python3 -m pytest -v --junitxml=pytest_compute.xml --durations=100 --ignore=tests/compute/test_ffi.py tests/compute || fail "compute"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment