diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index e740f6af6..8f8bf5c1f 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -177,7 +177,9 @@ jobs: sgl-kernel-build-wheels: needs: [check-changes, call-gate] - if: needs.check-changes.outputs.sgl_kernel == 'true' + if: | + !inputs.target_stage && + needs.check-changes.outputs.sgl_kernel == 'true' runs-on: x64-kernel-build-node strategy: matrix: @@ -218,7 +220,9 @@ jobs: sgl-kernel-build-wheels-arm: needs: [check-changes, call-gate] - if: needs.check-changes.outputs.sgl_kernel == 'true' + if: | + !inputs.target_stage && + needs.check-changes.outputs.sgl_kernel == 'true' runs-on: arm-kernel-build-node strategy: matrix: @@ -260,7 +264,9 @@ jobs: sgl-kernel-unit-test: needs: [check-changes, call-gate, sgl-kernel-build-wheels] - if: needs.check-changes.outputs.sgl_kernel == 'true' + if: | + !inputs.target_stage && + needs.check-changes.outputs.sgl_kernel == 'true' runs-on: 1-gpu-runner env: RUNNER_LABELS: 1-gpu-runner @@ -293,7 +299,9 @@ jobs: sgl-kernel-mla-test: needs: [check-changes, call-gate, sgl-kernel-build-wheels] - if: needs.check-changes.outputs.sgl_kernel == 'true' + if: | + !inputs.target_stage && + needs.check-changes.outputs.sgl_kernel == 'true' runs-on: 1-gpu-runner env: RUNNER_LABELS: 1-gpu-runner @@ -326,7 +334,9 @@ jobs: sgl-kernel-benchmark-test: needs: [check-changes, call-gate, sgl-kernel-build-wheels] - if: needs.check-changes.outputs.sgl_kernel == 'true' + if: | + !inputs.target_stage && + needs.check-changes.outputs.sgl_kernel == 'true' runs-on: 1-gpu-runner env: CI: true @@ -372,7 +382,9 @@ jobs: sgl-kernel-b200-test: needs: [check-changes, sgl-kernel-build-wheels] - if: needs.check-changes.outputs.sgl_kernel == 'true' + if: | + !inputs.target_stage && + needs.check-changes.outputs.sgl_kernel == 'true' runs-on: ${{ needs.check-changes.outputs.b200_runner }} env: RUNNER_LABELS: ${{ needs.check-changes.outputs.b200_runner }}