This is an automated email from the ASF dual-hosted git repository.
paleolimbot pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/arrow-nanoarrow.git
The following commit(s) were added to refs/heads/main by this push:
new 5f36032e chore: Re-enable CUDA tests with runs-on (#839)
5f36032e is described below
commit 5f36032edae49cda857bf1c985fa639c8b61bb44
Author: Dewey Dunnington <[email protected]>
AuthorDate: Wed Jan 14 07:07:40 2026 -0600
chore: Re-enable CUDA tests with runs-on (#839)
Closes #814.
---
.github/workflows/build-and-test-device.yaml | 14 ++++++++++----
1 file changed, 10 insertions(+), 4 deletions(-)
diff --git a/.github/workflows/build-and-test-device.yaml
b/.github/workflows/build-and-test-device.yaml
index 135b9739..e75bf310 100644
--- a/.github/workflows/build-and-test-device.yaml
+++ b/.github/workflows/build-and-test-device.yaml
@@ -50,9 +50,7 @@ jobs:
- {runner: ubuntu-latest, label: bundled-build, cmake_args:
"-DNANOARROW_DEVICE_BUNDLE=ON"}
- {runner: ubuntu-latest, label: shared-test-linkage, cmake_args:
"-DNANOARROW_TEST_LINKAGE_SHARED=ON"}
- {runner: macOS-latest, label: with-metal, cmake_args:
"-DNANOARROW_DEVICE_WITH_METAL=ON"}
- # This runner is currently not available
- # https://github.com/apache/arrow-nanoarrow/issues/814
- # - {runner: ["self-hosted", "cuda"], label: with-cuda, cmake_args:
"-DNANOARROW_DEVICE_WITH_CUDA=ON"}
+ - {runner: "runs-on=${{ github.run_id
}}/family=g4dn.xlarge/image=ubuntu24-gpu-x64/spot=capacity-optimized", label:
with-cuda, cmake_args: "-DNANOARROW_DEVICE_WITH_CUDA=ON"}
steps:
- uses: actions/checkout@v4
@@ -66,9 +64,17 @@ jobs:
if: matrix.config.label == 'with-cuda'
uses: actions/setup-python@v5
with:
- python-version: "3.x"
+ # cupy-cuda12x does not have a PyPI distribution for Python 3.14
+ python-version: "3.13"
cache: 'pip'
+ - name: Display NVIDIA SMI details
+ if: matrix.config.label == 'with-cuda'
+ run: |
+ nvidia-smi
+ nvidia-smi -L
+ nvidia-smi -q -d Memory
+
# The self-hosted runner for CUDA needs a number of packages installed
that
# are already installed on the GitHub runner
- name: Install build dependencies
