Skip to content

Commit 806b9a7

Browse files
committed
update CI handling of installing cuda.core dependencies when testing against local CTK
1 parent 310f391 commit 806b9a7

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

ci/tools/run-tests

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,13 @@ elif [[ "${test_module}" == "core" ]]; then
5757
echo "Installing core wheel"
5858
pwd
5959
ls
60-
pip install $(ls *.whl)["cu${TEST_CUDA_MAJOR}","test-cu${TEST_CUDA_MAJOR}"]
60+
if [[ "${LOCAL_CTK}" == 1 ]]; then
61+
# We already installed cuda-bindings, and all CTK components exist locally,
62+
# so just install the test dependencies.
63+
pip install $(ls *.whl)["test-cu${TEST_CUDA_MAJOR}"]
64+
else
65+
pip install $(ls *.whl)["cu${TEST_CUDA_MAJOR}","test-cu${TEST_CUDA_MAJOR}"]
66+
fi
6167
popd
6268
pushd ./cuda_core
6369
echo "Running core tests"
@@ -69,4 +75,4 @@ elif [[ "${test_module}" == "core" ]]; then
6975
${SANITIZER_CMD} pytest -rxXs -v tests/cython
7076
fi
7177
popd
72-
fi
78+
fi

0 commit comments

Comments
 (0)