From 6ff2fe6a6ad8e2af96739e381518786fb782b4ff Mon Sep 17 00:00:00 2001
From: "Po-Han Huang (NVIDIA)" <53919306+nvpohanh@users.noreply.github.com>
Date: Sat, 29 Aug 2026 00:32:58 +0800
Subject: [PATCH] CI: split JIT kernel unit tests into two partitions (#36775)

---
 .github/workflows/pr-test-jit-kernel.yml | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/pr-test-jit-kernel.yml b/.github/workflows/pr-test-jit-kernel.yml
index d4d039aeb..dfdb4cd72 100644
--- a/.github/workflows/pr-test-jit-kernel.yml
+++ b/.github/workflows/pr-test-jit-kernel.yml
@@ -47,6 +47,10 @@ jobs:
     # Runs whenever call-jit-kernel-tests dispatches this workflow. That caller is the
     # single gate (PR jit_kernel changes, or scheduled/parallel-dispatch full runs), so
     # the sub-jobs no longer re-exclude schedule/parallel-dispatch here.
+    strategy:
+      fail-fast: false
+      matrix:
+        partition: [0, 1]
     runs-on: 1-gpu-h100
     timeout-minutes: 240
     steps:
@@ -85,7 +89,10 @@ jobs:
         timeout-minutes: 30
         run: |
           cd test/
-          python3 run_suite.py --hw cuda --suite base-b-kernel-unit-test-1-gpu-large
+          python3 run_suite.py --hw cuda \
+            --suite base-b-kernel-unit-test-1-gpu-large \
+            --auto-partition-id ${{ matrix.partition }} \
+            --auto-partition-size 2
 
   jit-kernel-multigpu-unit-test:
     # Runs whenever call-jit-kernel-tests dispatches this workflow. That caller is the