From 39b9a7ce41057a75e1a3067d5250a6238043cff0 Mon Sep 17 00:00:00 2001
From: Evan Smothers <ebs@fb.com>
Date: Tue, 18 Jun 2024 15:52:07 -0700
Subject: [PATCH] Run unit tests on GPUs in CI

---
 .../{recipe_test_multi_gpu.yaml => gpu_test.yaml}      | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)
 rename .github/workflows/{recipe_test_multi_gpu.yaml => gpu_test.yaml} (80%)

diff --git a/.github/workflows/recipe_test_multi_gpu.yaml b/.github/workflows/gpu_test.yaml
similarity index 80%
rename from .github/workflows/recipe_test_multi_gpu.yaml
rename to .github/workflows/gpu_test.yaml
index d62bcc81bc..286b3fd4cc 100644
--- a/.github/workflows/recipe_test_multi_gpu.yaml
+++ b/.github/workflows/gpu_test.yaml
@@ -1,4 +1,4 @@
-name: Multi-GPU Recipe Tests
+name: GPU tests
 
 on:
   push:
@@ -7,7 +7,7 @@ on:
   workflow_dispatch:
 
 concurrency:
-  group: recipe-test-multi-gpu-${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && github.run_number || github.ref }}
+  group: gpu-test-${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && github.run_number || github.ref }}
   cancel-in-progress: true
 
 permissions:
@@ -19,7 +19,7 @@ defaults:
     shell: bash -l -eo pipefail {0}
 
 jobs:
-  recipe_test_multi_gpu:
+  gpu_test:
     runs-on: linux.8xlarge.nvidia.gpu
     strategy:
       matrix:
@@ -47,7 +47,7 @@ jobs:
         run: |
           python -m pip install -e ".[dev]"
           python -m pip install lm-eval==0.4.*
-      - name: Run recipe tests with coverage
-        run: pytest tests -m integration_test --cov=. --cov-report=xml --durations=20 -vv
+      - name: Run recipe and unit tests with coverage
+        run: pytest tests --with-integration --cov=. --cov-report=xml --durations=20 -vv
       - name: Upload Coverage to Codecov
         uses: codecov/codecov-action@v3