Add thirdparty tests of Awkward Array

Author: gmarkall

.github/workflows/pr.yaml

@@ -33,6 +33,7 @@ jobs:
       - test-wheels-deps-wheels
       - test-thirdparty-cudf
       - test-thirdparty-nvmath
+      - test-thirdparty-awkward
       - build-docs
       - coverage-report
     secrets: inherit
@@ -150,6 +151,18 @@ jobs:
       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))]))
+  test-thirdparty-awkward:
+    needs:
+      - build-wheels
+      - compute-matrix
+    uses: ./.github/workflows/wheels-test.yaml
+    with:
+      build_type: pull-request
+      script: "ci/test_thirdparty_awkward.sh"
+      matrix: ${{ needs.compute-matrix.outputs.TEST_MATRIX }}
+      # TODO: Enable for CUDA 13 in future
+      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_awkward.sh

@@ -0,0 +1,42 @@
+#!/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%.*}
+AWKWARD_VERSION="2.8.10"
+
+rapids-logger "Install awkward and related libraries"
+
+pip install awkward==${AWKWARD_VERSION} cupy pyarrow pandas nox
+
+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 "Clone awkward repository"
+git clone --recursive https://github.com/sckit-hep/awkward.git
+pushd awkward
+git checkout v${AWKWARD_VERSION}
+
+rapids-logger "Generate awkward tests"
+nox -s prepare -- --tests
+
+rapids-logger "Check GPU usage"
+nvidia-smi
+
+rapids-logger "Show Numba system info"
+python -m numba --sysinfo
+
+rapids-logger "Run Awkward CUDA tests"
+python -m pytest -v -n auto tests-cuda tests-cuda-kernels tests-cuda-kernels-explicit
+
+popd