Update CI to use new workflow and silicon.

huggingface · Jul 7, 2024 · 89c4b88 · 89c4b88
1 parent 011b5a9
commit 89c4b88
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 78 deletions.
diff --git a/.github/workflows/pr_functional_tests.yml b/.github/workflows/pr_functional_tests.yml
diff --git a/.github/workflows/pr_integration_tests.yml → .github/workflows/pr_tests.yml b/.github/workflows/pr_integration_tests.yml → .github/workflows/pr_tests.yml
@@ -14,9 +14,6 @@ concurrency:
 
 env:
   OPTIMUM_NVIDIA_IS_CI: ON
-  RUN_CPU_ONLY: OFF
-  RUN_NIGHTLY: OFF
-  RUN_SLOW: ON
   HF_HUB_ENABLE_HF_TRANSFER: ON
 
 jobs:
@@ -25,12 +22,13 @@ jobs:
       fail-fast: false
       matrix:
         config:
-          - name: GPU-enabled Optimum-Nvidia Integration Test Suite
-            image: nvidia/cuda:12.1.0-devel-ubuntu22.04
-        cuda_arch: [sm_86, sm_89]
+          - name: GPU-enabled Optimum-Nvidia Test Suite
+            image: nvidia/cuda:12.4.1-devel-ubuntu22.04
+        gpu_target: ["nvidia-multi-gpu-l4-runners", "nvidia-multi-gpu-a10-runners"]
 
     name: ${{ matrix.config.name }}
-    runs-on: [ci, nvidia-gpu, multi-gpu, "${{ matrix.cuda_arch }}"]
+    runs-on:
+      group: "${{matrix.gpu_target}}"
 
     container:
       image: ${{ matrix.config.image }}
@@ -65,6 +63,6 @@ jobs:
         run: |
           python -c "from tensorrt_llm import __version__; print(__version__)"
 
-      - name: Run optimum-nvidia integration test-suite
+      - name: Run optimum-nvidia test-suite
         run: |
-          pytest -s -vvvvv -n 1 -p no:warnings -o log_cli=true tests/integration/
+          pytest -s -vvvvv -n 4 -p no:warnings -o log_cli=true --ignore=tests/integration tests/