From 124099c382a437fd49affb75727f1c45743cc62a Mon Sep 17 00:00:00 2001
From: Paul Zhang <paulzhan@umich.edu>
Date: Mon, 30 Sep 2024 21:03:02 +0000
Subject: [PATCH] Update gpu ci2

---
 .github/workflows/unittest_ci.yml | 69 +++++++++++++++----------------
 1 file changed, 34 insertions(+), 35 deletions(-)

diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml
index a0f4ca503..6598701fd 100644
--- a/.github/workflows/unittest_ci.yml
+++ b/.github/workflows/unittest_ci.yml
@@ -12,42 +12,41 @@ on:
 jobs:
   build_test:
     strategy:
+      fail-fast: false
       matrix:
-        os: [linux.g5.12xlarge.nvidia.gpu]
-        matrix:
-          include:
-            - os: linux.g5.12xlarge.nvidia.gpu
-              python-version: 3.9
-              python-tag: "py39"
-              cuda-tag: "cu118"
-            - os: linux.g5.12xlarge.nvidia.gpu
-              python-version: 3.9
-              python-tag: "py39"
-              cuda-tag: "cu121"
-            - os: linux.g5.12xlarge.nvidia.gpu
-              python-version: '3.10'
-              python-tag: "py310"
-              cuda-tag: "cu118"
-            - os: linux.g5.12xlarge.nvidia.gpu
-              python-version: '3.10'
-              python-tag: "py310"
-              cuda-tag: "cu121"
-            - os: linux.g5.12xlarge.nvidia.gpu
-              python-version: '3.11'
-              python-tag: "py311"
-              cuda-tag: "cu118"
-            - os: linux.g5.12xlarge.nvidia.gpu
-              python-version: '3.11'
-              python-tag: "py311"
-              cuda-tag: "cu121"
-            - os: linux.g5.12xlarge.nvidia.gpu
-              python-version: '3.12'
-              python-tag: "py312"
-              cuda-tag: "cu118"
-            - os: linux.g5.12xlarge.nvidia.gpu
-              python-version: '3.12'
-              python-tag: "py312"
-              cuda-tag: "cu121"
+        include:
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: 3.9
+            python-tag: "py39"
+            cuda-tag: "cu118"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: 3.9
+            python-tag: "py39"
+            cuda-tag: "cu121"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.10'
+            python-tag: "py310"
+            cuda-tag: "cu118"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.10'
+            python-tag: "py310"
+            cuda-tag: "cu121"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.11'
+            python-tag: "py311"
+            cuda-tag: "cu118"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.11'
+            python-tag: "py311"
+            cuda-tag: "cu121"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.12'
+            python-tag: "py312"
+            cuda-tag: "cu118"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.12'
+            python-tag: "py312"
+            cuda-tag: "cu121"
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: ${{ matrix.os }}