Add thirdparty tests of numba extensions

.github/workflows/pr.yaml

@@ -25,6 +25,7 @@ jobs:
       - test-wheels
       - test-wheels-ctypes-binding
       - test-wheels-deps-wheels
+      - test-thirdparty
       - build-docs
     secrets: inherit
     uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@branch-25.04
@@ -135,6 +136,16 @@ jobs:
       script: "ci/test_wheel_deps_wheels.sh"
       matrix: ${{ needs.compute-matrix.outputs.TEST_MATRIX }}
       matrix_filter: map(select(.ARCH == "amd64" and (.CUDA_VER | split(".") | .[0] | tonumber >= 12))) | group_by(.CUDA_VER|split(".")|map(tonumber)|.[0]) | map(max_by([(.PY_VER|split(".")|map(tonumber)), (.CUDA_VER|split(".")|map(tonumber))]))
+  test-thirdparty:
+    needs:
+      - build-wheels
+      - compute-matrix
+    uses: ./.github/workflows/wheels-test.yaml
+    with:
+      build_type: pull-request
+      script: "ci/test_thirdparty.sh"
+      matrix: ${{ needs.compute-matrix.outputs.TEST_MATRIX }}
+      matrix_filter: map(select(.ARCH == "amd64" and (.CUDA_VER | split(".") | .[0] | tonumber >= 12))) | group_by(.CUDA_VER|split(".")|map(tonumber)|.[0]) | map(max_by([(.PY_VER|split(".")|map(tonumber)), (.CUDA_VER|split(".")|map(tonumber))]))
   build-docs:
     needs:
       - build-conda

ci/test_thirdparty.sh

@@ -0,0 +1,40 @@
+#!/bin/bash
+# Copyright (c) 2025, NVIDIA CORPORATION
+
+set -euo pipefail
+
+CUDA_VER_MAJOR_MINOR=${CUDA_VER%.*}
+
+rapids-logger "Install wheel with test dependencies"
+package=$(realpath wheel/numba_cuda*.whl)
+echo "Package path: ${package}"
+python -m pip install \
+    "${package}[test]" \
+    "cuda-python==${CUDA_VER_MAJOR_MINOR%.*}.*" \
+    "cuda-core==0.3.*" \
+
+
+rapids-logger "Install cuDF Wheel"
+pip install \
+    --extra-index-url=https://pypi.nvidia.com \
+    "cudf-cu12==25.6.*"
+
+rapids-logger "Shallow clone cuDF repository"
+git clone --single-branch --branch 'branch-25.06' https://github.com/rapidsai/cudf.git
+
+pushd cudf
+
+rapids-logger "Check GPU usage"
+nvidia-smi
+
+
+rapids-logger "Show Numba system info"
+python -m numba --sysinfo
+
+rapids-logger "Run Scalar UDF tests"
+py.test python/cudf/cudf/tests/test_udf_masked_ops.py
+
+rapids-logger "Run GroupBy UDF tests"
+py.test python/cudf/cudf/tests/test_groupby.py -k test_groupby_apply_jit
+
+popd