From 21edcfe09580786300db5d653c9f38124b6d91f6 Mon Sep 17 00:00:00 2001
From: Lin Tian <54332303+lintian-a@users.noreply.github.com>
Date: Wed, 31 Jul 2024 23:03:29 +0800
Subject: [PATCH] Create gpu-test-action.yml

---
 .github/workflows/gpu-test-action.yml | 31 +++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 .github/workflows/gpu-test-action.yml

diff --git a/.github/workflows/gpu-test-action.yml b/.github/workflows/gpu-test-action.yml
new file mode 100644
index 0000000..f620712
--- /dev/null
+++ b/.github/workflows/gpu-test-action.yml
@@ -0,0 +1,31 @@
+name: gpu-tests
+
+on:
+  pull_request:
+  push:
+    branches: master
+
+jobs:
+  test-linux:
+    runs-on: [self-hosted, linux]
+    strategy:
+      max-parallel: 5
+
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python 3.7
+      uses: actions/setup-python@v2
+      with:
+        python-version: 3.7
+    - name: Install dependencies
+      run: |
+        pip install -r requirements.txt
+        
+        pip install -e .
+
+    - name: fast test with unittest
+      run: |
+        python -m unittest -k CPU
+    - name: GPU test with unittest
+      run: |
+        python -m unittest discover