From 1a3d6f31da58f37d7bb63e25f2b7dafd4765d75b Mon Sep 17 00:00:00 2001
From: hzh0425 <hzh0425@apache.org>
Date: Sat, 6 Sep 2025 10:28:42 +0800
Subject: [PATCH] Modify ci workflow for auto-partitioning in 2-GPU backend
 tests (#10029)

---
 .github/workflows/pr-test.yml | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index 153eb22e7..bd1053902 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -90,6 +90,10 @@ jobs:
         github.event.pull_request.draft == false &&
         needs.check-changes.outputs.src == 'true'
     runs-on: 2-gpu-runner
+    strategy:
+      fail-fast: false
+      matrix:
+        part: [0, 1]
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
@@ -102,7 +106,7 @@ jobs:
         timeout-minutes: 30
         run: |
           cd test/srt
-          python3 run_suite.py --suite per-commit-2-gpu
+          python3 run_suite.py --suite per-commit-2-gpu --auto-partition-id ${{ matrix.part }} --auto-partition-size 2
 
   unit-test-backend-4-gpu:
     needs: [check-changes, unit-test-backend-2-gpu]