From 17af39c5dc6a20f39d5a68dd1ac668477eacadce Mon Sep 17 00:00:00 2001
From: Yineng Zhang <me@zhyncs.com>
Date: Tue, 30 Jul 2024 17:32:13 +1000
Subject: [PATCH] feat: add runner (#821)

---
 .github/workflows/pr-e2e-test.yml | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 .github/workflows/pr-e2e-test.yml

diff --git a/.github/workflows/pr-e2e-test.yml b/.github/workflows/pr-e2e-test.yml
new file mode 100644
index 000000000..7f6a58407
--- /dev/null
+++ b/.github/workflows/pr-e2e-test.yml
@@ -0,0 +1,30 @@
+name: PR E2E Test
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+  workflow_dispatch:
+
+jobs:
+  gpu-job:
+    runs-on: self-hosted
+    env:
+      CUDA_VISIBLE_DEVICES: 6
+    steps:
+    - uses: actions/checkout@v2
+    - name: Check GPU
+      run: |
+        if ! command -v nvidia-smi &> /dev/null; then
+          echo "nvidia-smi not found. Is CUDA installed?"
+          exit 1
+        fi
+        nvidia-smi || exit 1
+    - name: Environment Info
+      run: |
+        echo "Working directory: $(pwd)"
+        echo "CUDA_VISIBLE_DEVICES: $CUDA_VISIBLE_DEVICES"
+    - name: Run Tests
+      run: |
+        echo "Running tests..."