Fix sgl-kernel jobs to skip when target_stage is specified (#16308)

This commit is contained in:
Alison Shao
2026-01-02 12:49:26 -08:00
committed by GitHub
parent d401d23876
commit f0195627a9

View File

@@ -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 }}