From d41d19be46c083db83e8b302ed2b982b8ec25132 Mon Sep 17 00:00:00 2001
From: edknv <edwardk@nvidia.com>
Date: Tue, 17 Oct 2023 10:12:46 -0700
Subject: [PATCH] Add GPU CI workflow

---
 .github/workflows/gpu-ci.yml | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 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..7cbcc54
--- /dev/null
+++ b/.github/workflows/gpu-ci.yml
@@ -0,0 +1,30 @@
+name: GPU CI
+
+on:
+  workflow_dispatch:
+  push:
+    branches:
+      - main
+      - "pull-request/[0-9]+"
+    tags:
+      - "v[0-9]+.[0-9]+.[0-9]+"
+
+jobs:
+  gpu-ci:
+    runs-on: linux-amd64-gpu-p100-latest-1
+    container:
+      image: nvcr.io/nvidia/pytorch:23.09-py3
+      env:
+        NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
+      options: --shm-size=1G
+      credentials:
+        username: $oauthtoken
+        password: ${{ secrets.NGC_TOKEN }}
+
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+      - name: Run tests
+        run: |
+          pytest tests/