diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 794bcce19..727f2e2e9 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -791,34 +791,34 @@ jobs: cd test/srt python3 run_suite.py --suite per-commit-8-gpu-h200-deepseek-v32 - # unit-test-backend-4-gpu-b200: - # needs: [check-changes, unit-test-backend-2-gpu, sgl-kernel-build-wheels] - # if: always() && !failure() && !cancelled() && - # ((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true')) - # runs-on: 4-gpu-b200 - # strategy: - # fail-fast: false - # steps: - # - name: Checkout code - # uses: actions/checkout@v4 + unit-test-backend-4-gpu-b200: + needs: [check-changes, unit-test-backend-2-gpu, sgl-kernel-build-wheels] + if: always() && !failure() && !cancelled() && + ((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true')) + runs-on: 4-gpu-b200 + strategy: + fail-fast: false + steps: + - name: Checkout code + uses: actions/checkout@v4 - # - name: Download artifacts - # if: needs.check-changes.outputs.sgl_kernel == 'true' - # uses: actions/download-artifact@v4 - # with: - # path: sgl-kernel/dist/ - # merge-multiple: true - # pattern: wheel-python3.10-cuda12.9 + - name: Download artifacts + if: needs.check-changes.outputs.sgl_kernel == 'true' + uses: actions/download-artifact@v4 + with: + path: sgl-kernel/dist/ + merge-multiple: true + pattern: wheel-python3.10-cuda12.9 - # - name: Install dependencies - # run: | - # CUSTOM_BUILD_SGL_KERNEL=${{needs.check-changes.outputs.sgl_kernel}} IS_BLACKWELL=1 bash scripts/ci/ci_install_dependency.sh + - name: Install dependencies + run: | + CUSTOM_BUILD_SGL_KERNEL=${{needs.check-changes.outputs.sgl_kernel}} IS_BLACKWELL=1 bash scripts/ci/ci_install_dependency.sh - # - name: Run test - # timeout-minutes: 45 - # run: | - # cd test/srt - # python3 run_suite.py --suite per-commit-4-gpu-b200 --auto-partition-id 0 --auto-partition-size 1 --timeout-per-file 3600 + - name: Run test + timeout-minutes: 45 + run: | + cd test/srt + python3 run_suite.py --suite per-commit-4-gpu-b200 --auto-partition-id 0 --auto-partition-size 1 --timeout-per-file 3600 unit-test-backend-4-gpu-gb200: needs: [check-changes, sgl-kernel-build-wheels-arm] @@ -872,7 +872,7 @@ jobs: accuracy-test-2-gpu, unit-test-deepep-4-gpu, unit-test-deepep-8-gpu, - # unit-test-backend-4-gpu-b200, + unit-test-backend-4-gpu-b200, unit-test-backend-4-gpu-gb200, ] if: always()