Test GPU runner

.github/workflows/test.yml

@@ -1,65 +1,15 @@
-name: Merge group tests
+name: GPU CI
 
 on:
-  merge_group:
-  pull_request:
+  push:
+    branches:
+      - gpu-ci
 
 jobs:
-  test-ref:
-    runs-on: ubuntu-latest
-    if: github.event_name == 'merge_group'
+  gpu-ci:
+    runs-on: ubuntu-gpu
     steps:
       - uses: actions/checkout@v4
-      - name: Print branch names
+      - name: Test GPU
         run: |
-            echo "Head ref: ${{ github.event.merge_group.head_ref }}"
-            echo "Base ref: ${{ github.event.merge_group.base_ref }}"
-      - uses: actions/checkout@v4
-        with:
-          ref: ${{ github.event.merge_group.base_ref }}
-
-  test-ref-b:
-    runs-on: ubuntu-latest
-    if: github.event_name == 'pull_request'
-    steps:
-      - uses: actions/checkout@v4
-      - name: Print branch name
-        run: |
-            echo "Base ref: ${{ github.base_ref }}"
-
-  check-job-results:
-    if: always()
-    needs: [test-ref, test-ref-b]
-    runs-on: ubuntu-latest
-    steps:
-      - name: Check job results
-        id: check-results
-        run: |
-          # Create an associative array of job statuses
-          declare -A job_results=(
-            ["test-ref"]="${{ needs.test-ref.result }}"
-            ["test-ref-b"]="${{ needs.test-ref-b.result }}"
-          )
-
-          # Iterate through jobs and get their results
-          failed_count=0
-          for job in "${!job_results[@]}"; do
-            RESULT=${job_results[$job]}
-            if [[ "$RESULT" == "failure" ]]; then
-              failed_count=$((failed_count + 1))
-            fi
-            echo "$job result: $RESULT"
-          done
-
-          if [ "$failed_count" -gt 0 ]; then
-            echo "Some jobs failed"
-            echo "result=failure" | tee -a $GITHUB_OUTPUT
-          else
-            echo "All jobs succeeded or were skipped"
-            echo "result=success" | tee -a $GITHUB_OUTPUT
-          fi
-      - name: Error on job failure
-        run: |
-          if [[ "${{ steps.check-results.outputs.result }}" == "failure" ]]; then
-            exit 1
-          fi
+          nvidia-smi