Add nvmath-python tests

Author: gmarkall

.github/workflows/pr.yaml

@@ -32,6 +32,7 @@ jobs:
       - test-wheels
       - test-wheels-deps-wheels
       - test-thirdparty-cudf
+      - test-thirdparty-nvmath
       - build-docs
       - coverage-report
     secrets: inherit
@@ -138,6 +139,17 @@ jobs:
       matrix: ${{ needs.compute-matrix.outputs.TEST_MATRIX }}
       # TODO: Enable for CUDA 13 when a supporting version of cuDF is available
       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-nvmath:
+    needs:
+      - build-wheels
+      - compute-matrix
+    uses: ./.github/workflows/wheels-test.yaml
+    with:
+      build_type: pull-request
+      script: "ci/test_thirdparty_nvmath.sh"
+      matrix: ${{ needs.compute-matrix.outputs.TEST_MATRIX }}
+      # TODO: Enable for CUDA 13 when a supporting version of nvmath-python is available
+      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:
     uses: ./.github/workflows/docs-build.yaml
   coverage-report:

ci/test_thirdparty_nvmath.sh

@@ -0,0 +1,49 @@
+#!/bin/bash
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+set -euo pipefail
+
+CUDA_VER_MAJOR_MINOR=${CUDA_VER%.*}
+
+NVMATH_PYTHON_VERSION="0.6.*"
+
+rapids-logger "Install nvmath-python"
+
+pip install nvmath-python[cu12,dx]==${NVMATH_PYTHON_VERSION}
+
+rapids-logger "Remove Extraneous numba-cuda"
+pip uninstall -y numba-cuda
+
+rapids-logger "Install wheel with test dependencies"
+package=$(realpath wheel/numba_cuda*.whl)
+echo "Package path: ${package}"
+python -m pip install \
+    "${package}" \
+    "cuda-python==${CUDA_VER_MAJOR_MINOR%.*}.*" \
+    "cuda-core==0.3.*" \
+    "nvidia-nvjitlink-cu12" \
+    --group test
+
+
+rapids-logger "Shallow clone nvmath-python repository"
+git clone --single-branch --branch 'release-0.6.x' https://github.com/NVIDIA/nvmath-python.git
+
+rapids-logger "Install nvmath-python test dependencies"
+
+pip install nvmath-python/requirements/pip/tests.txt
+
+pushd nvmath-python/tests
+
+rapids-logger "Check GPU usage"
+nvidia-smi
+
+rapids-logger "Show Numba system info"
+python -m numba --sysinfo
+
+rapids-logger "Run nvmath-python device tests"
+# Required for nvmath-python to locate pip-install MathDx
+export MATHDX_HOME=${CONDA_PREFIX}/lib/python3.13/site-packages/nvidia/mathdx
+python -m pytest nvmath_tests/device
+
+popd