[MUSA] Harden CI dependencies and diffusion warmup (#35610)

This commit is contained in:
R0CKSTAR
2026-08-20 09:40:24 -07:00
committed by GitHub
parent be373395b4
commit 81df6f2c57
6 changed files with 542 additions and 23 deletions
+5 -6
View File
@@ -66,7 +66,7 @@ jobs:
- name: Install dependencies
if: steps.gate.outputs.run_job == 'true'
timeout-minutes: 15
timeout-minutes: 30
run: |
bash scripts/ci/musa/musa_install_dependency.sh
@@ -74,7 +74,6 @@ jobs:
if: steps.gate.outputs.run_job == 'true'
timeout-minutes: 30
run: |
pytest python/sglang/kernels/aot/tests/test_per_token_quant_fp8.py
pytest python/sglang/kernels/aot/tests/speculative/test_eagle_utils.py
pytest python/sglang/kernels/aot/tests/speculative/test_ngram_utils.py
pytest python/sglang/kernels/aot/tests/speculative/test_speculative_sampling.py
@@ -111,7 +110,7 @@ jobs:
ref: ${{ inputs.ref || github.ref }}
- name: Install dependencies
timeout-minutes: 15
timeout-minutes: 30
if: steps.gate.outputs.run_job == 'true'
run: |
bash scripts/ci/musa/musa_install_dependency.sh
@@ -154,7 +153,7 @@ jobs:
ref: ${{ inputs.ref || github.ref }}
- name: Install dependencies
timeout-minutes: 15
timeout-minutes: 30
if: steps.gate.outputs.run_job == 'true'
run: |
bash scripts/ci/musa/musa_install_dependency.sh
@@ -206,7 +205,7 @@ jobs:
ref: ${{ inputs.ref || github.ref }}
- name: Install dependencies
timeout-minutes: 15
timeout-minutes: 30
if: steps.gate.outputs.run_job == 'true'
run: |
bash scripts/ci/musa/musa_install_dependency.sh
@@ -255,7 +254,7 @@ jobs:
ref: ${{ inputs.ref || github.ref }}
- name: Install dependencies
timeout-minutes: 15
timeout-minutes: 30
if: steps.gate.outputs.run_job == 'true'
run: |
bash scripts/ci/musa/musa_install_dependency.sh
+10 -6
View File
@@ -72,7 +72,7 @@ jobs:
- "python/pyproject_other.toml"
- "python/sglang/!(multimodal_gen)/**"
- "python/sglang/srt/hardware_backend/musa/**"
- "scripts/ci/musa/*"
- "scripts/ci/musa/**"
- "scripts/ci/utils/*"
multimodal_gen:
- "python/pyproject_other.toml"
@@ -81,9 +81,14 @@ jobs:
- "python/sglang/multimodal_gen/test/unit/musa/layers/test_musa_silu_and_mul.py"
- "python/sglang/multimodal_gen/test/server/musa/run_suite.py"
- "python/sglang/multimodal_gen/test/server/musa/**"
- "scripts/ci/musa/**"
sgl_kernel:
- ".github/workflows/pr-test-musa.yml"
- "python/sglang/kernels/aot/csrc/musa/**"
- "scripts/ci/musa/**"
- name: Test MUSA dependency constraints
run: python3 -m unittest scripts/ci/musa/test_musa_python_stack.py
# ==================== PR Gate ==================== #
pr-gate:
@@ -117,7 +122,7 @@ jobs:
ref: ${{ inputs.ref || github.ref }}
- name: Install dependencies
timeout-minutes: 15
timeout-minutes: 30
run: |
bash scripts/ci/musa/musa_install_dependency.sh
@@ -152,7 +157,7 @@ jobs:
ref: ${{ inputs.ref || github.ref }}
- name: Install dependencies
timeout-minutes: 15
timeout-minutes: 30
run: |
bash scripts/ci/musa/musa_install_dependency.sh
@@ -183,7 +188,7 @@ jobs:
uses: actions/checkout@v4
- name: Install dependencies
timeout-minutes: 15
timeout-minutes: 30
run: |
bash scripts/ci/musa/musa_install_dependency.sh
@@ -210,14 +215,13 @@ jobs:
ref: ${{ inputs.ref || github.ref }}
- name: Install dependencies
timeout-minutes: 15
timeout-minutes: 30
run: |
bash scripts/ci/musa/musa_install_dependency.sh
- name: Run sgl-kernel test
timeout-minutes: 20
run: |
pytest python/sglang/kernels/aot/tests/test_per_token_quant_fp8.py
pytest python/sglang/kernels/aot/tests/speculative/test_eagle_utils.py
pytest python/sglang/kernels/aot/tests/speculative/test_ngram_utils.py
pytest python/sglang/kernels/aot/tests/speculative/test_speculative_sampling.py