Skip to content

Commit 673016b

Browse files
committed
Narrow CI workaround for cufile ABI issue
1 parent e253173 commit 673016b

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

‎ci/tools/run-tests‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,11 @@ elif [[ "${test_module}" == "bindings" ]]; then
5555
if [[ "${LOCAL_CTK}" == 1 ]]; then
5656
pip install "${CUDA_BINDINGS_ARTIFACTS_DIR}"/*.whl --group test "${TEST_FT_GROUP[@]}"
5757
else
58-
pip install $(ls "${CUDA_BINDINGS_ARTIFACTS_DIR}"/*.whl)[all] --group test "${TEST_FT_GROUP[@]}"
58+
TEST_CUDA_MAJOR_MINOR="$(cut -d '.' -f 1-2 <<< "${CUDA_VER}")"
59+
# Keep the wheel-installed Toolkit aligned with the matrix-selected minor;
60+
# newer minors can grow ABI structs beyond the size compiled into this wheel.
61+
pip install $(ls "${CUDA_BINDINGS_ARTIFACTS_DIR}"/*.whl)[all] --group test "${TEST_FT_GROUP[@]}" \
62+
"cuda-toolkit==${TEST_CUDA_MAJOR_MINOR}.*"
5963
fi
6064
echo "Running bindings tests"
6165
${SANITIZER_CMD} pytest -rxXs -v --randomly-dont-reorganize "${PYTEST_PARALLEL_ARGS[@]}" tests/

0 commit comments

Comments
 (0)