diff --git a/ci/run_pylibcudf_pytests.sh b/ci/run_pylibcudf_pytests.sh index 171be767a01..b4ba91dfdc1 100755 --- a/ci/run_pylibcudf_pytests.sh +++ b/ci/run_pylibcudf_pytests.sh @@ -1,5 +1,5 @@ #!/bin/bash -# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION. +# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 set -euo pipefail @@ -39,8 +39,10 @@ EOF # Support invoking run_cudf_pytests.sh outside the script directory cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/pylibcudf/ -if [ -n "$TESTING_LIB" ] && [ -f "$TESTING_LIB" ]; then - # If the stream testing library was found, split the tests into two passes. - LD_PRELOAD="$TESTING_LIB" PYLIBCUDF_STREAM_TESTING=1 pytest --cache-clear -m "not uses_custom_stream" --ignore="benchmarks" "$@" tests +# If the stream testing library was found, split the tests into two passes. +echo "RAPIDS_CUDA_VERSION: $RAPIDS_CUDA_VERSION" +if [ -n "$TESTING_LIB" ] && [ -f "$TESTING_LIB" ] && [ "$RAPIDS_CUDA_VERSION" != "12.2.2" ]; then + # run the stream tests without xdist to ease the burden on the compute-sanitizer checks + LD_PRELOAD="$TESTING_LIB" PYLIBCUDF_STREAM_TESTING=1 pytest --cache-clear -m "not uses_custom_stream" --ignore="benchmarks" -p no:xdist tests fi pytest --cache-clear --ignore="benchmarks" "$@" tests