From 6544743f231635ebd56ca28cca3996c00f93544e Mon Sep 17 00:00:00 2001
From: Samuel Burnham <45365069+samuelburnham@users.noreply.github.com>
Date: Fri, 5 Jan 2024 14:37:48 -0500
Subject: [PATCH] Test GPU CI

---
 .github/workflows/gpu-ci.yml | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 .github/workflows/gpu-ci.yml

diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml
new file mode 100644
index 0000000..8d464cd
--- /dev/null
+++ b/.github/workflows/gpu-ci.yml
@@ -0,0 +1,35 @@
+# Runs the test suite on a self-hosted GPU machine with CUDA enabled
+name: GPU tests
+
+on:
+  pull_request:
+    types: [opened, synchronize, reopened, ready_for_review]
+    branches: [dev]
+  merge_group:
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
+  cancel-in-progress: true
+
+jobs:
+  cuda:
+    name: Rust tests on CUDA
+    if: github.event_name != 'pull_request' || github.event.action == 'enqueued'
+    runs-on: [self-hosted, gpu-ci]
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          repository: lurk-lab/ci-workflows
+      - uses: ./.github/actions/gpu-setup
+        with:
+          gpu-framework: 'cuda'
+      - uses: ./.github/actions/ci-env
+      - uses: actions/checkout@v4
+        with:
+          submodules: recursive
+      - uses: dtolnay/rust-toolchain@stable
+      - uses: taiki-e/install-action@nextest
+      - uses: Swatinem/rust-cache@v2
+      - name: CUDA tests
+        run: |
+          cargo nextest run --release --features dont-implement-sort