[diffusion] CI: refactor CI (#28762)
This commit is contained in:
@@ -121,8 +121,8 @@ jobs:
|
||||
if: steps.gate.outputs.run_job == 'true'
|
||||
timeout-minutes: 30
|
||||
run: |
|
||||
pytest python/sglang/multimodal_gen/test/layers/test_musa_rmsnorm.py
|
||||
pytest python/sglang/multimodal_gen/test/layers/test_musa_silu_and_mul.py
|
||||
pytest python/sglang/multimodal_gen/test/unit/musa/layers/test_musa_rmsnorm.py
|
||||
pytest python/sglang/multimodal_gen/test/unit/musa/layers/test_musa_silu_and_mul.py
|
||||
|
||||
# ==================== LLM server: 1-GPU ====================
|
||||
nightly-test-llm-server-1-gpu-musa:
|
||||
@@ -219,7 +219,7 @@ jobs:
|
||||
RUNAI_STREAMER_MEMORY_LIMIT: 0
|
||||
run: |
|
||||
cd python
|
||||
python3 sglang/multimodal_gen/test/run_suite_musa.py \
|
||||
python3 sglang/multimodal_gen/test/server/musa/run_suite.py \
|
||||
--suite 1-gpu-musa-nightly \
|
||||
--partition-id ${{ matrix.part }} \
|
||||
--total-partitions 2 \
|
||||
@@ -268,7 +268,7 @@ jobs:
|
||||
RUNAI_STREAMER_MEMORY_LIMIT: 0
|
||||
run: |
|
||||
cd python
|
||||
python3 sglang/multimodal_gen/test/run_suite_musa.py \
|
||||
python3 sglang/multimodal_gen/test/server/musa/run_suite.py \
|
||||
--suite 2-gpu-musa \
|
||||
--continue-on-error
|
||||
|
||||
|
||||
@@ -746,7 +746,7 @@ jobs:
|
||||
fail-fast: false
|
||||
matrix:
|
||||
runner: [linux-mi325-2gpu-sglang]
|
||||
part: [0, 1, 2] # 3 partitions: 2 parametrized + 1 standalone (test_disagg_server.py)
|
||||
part: [0, 1, 2] # 3 partitions: 2 parametrized + 1 standalone (single_test_file/test_disagg_server.py)
|
||||
runs-on: ${{matrix.runner}}
|
||||
steps:
|
||||
- name: Checkout code
|
||||
|
||||
@@ -767,7 +767,7 @@ jobs:
|
||||
strategy:
|
||||
fail-fast: false
|
||||
matrix:
|
||||
part: [0, 1, 2] # 3 partitions: 2 parametrized + 1 standalone (test_disagg_server.py)
|
||||
part: [0, 1, 2] # 3 partitions: 2 parametrized + 1 standalone (single_test_file/test_disagg_server.py)
|
||||
runs-on: ${{ format('linux-{0}-2gpu-sglang', inputs.runner_arch || 'mi325') }}
|
||||
steps:
|
||||
- name: Checkout code
|
||||
|
||||
@@ -77,9 +77,9 @@ jobs:
|
||||
multimodal_gen:
|
||||
- "python/pyproject_other.toml"
|
||||
- "python/sglang/multimodal_gen/runtime/platforms/musa.py"
|
||||
- "python/sglang/multimodal_gen/test/layers/test_musa_rmsnorm.py"
|
||||
- "python/sglang/multimodal_gen/test/layers/test_musa_silu_and_mul.py"
|
||||
- "python/sglang/multimodal_gen/test/run_suite_musa.py"
|
||||
- "python/sglang/multimodal_gen/test/unit/musa/layers/test_musa_rmsnorm.py"
|
||||
- "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/**"
|
||||
sgl_kernel:
|
||||
- ".github/workflows/pr-test-musa.yml"
|
||||
@@ -127,7 +127,7 @@ jobs:
|
||||
RUNAI_STREAMER_MEMORY_LIMIT: 0
|
||||
run: |
|
||||
cd python
|
||||
python3 sglang/multimodal_gen/test/run_suite_musa.py \
|
||||
python3 sglang/multimodal_gen/test/server/musa/run_suite.py \
|
||||
--suite 1-gpu-musa \
|
||||
--partition-id ${{ matrix.part }} \
|
||||
--total-partitions 2
|
||||
@@ -162,7 +162,7 @@ jobs:
|
||||
RUNAI_STREAMER_MEMORY_LIMIT: 0
|
||||
run: |
|
||||
cd python
|
||||
python3 sglang/multimodal_gen/test/run_suite_musa.py \
|
||||
python3 sglang/multimodal_gen/test/server/musa/run_suite.py \
|
||||
--suite 2-gpu-musa
|
||||
|
||||
multimodal-gen-layer-unit-test-musa:
|
||||
@@ -190,8 +190,8 @@ jobs:
|
||||
- name: Run multimodal gen layer unit test
|
||||
timeout-minutes: 30
|
||||
run: |
|
||||
pytest python/sglang/multimodal_gen/test/layers/test_musa_rmsnorm.py
|
||||
pytest python/sglang/multimodal_gen/test/layers/test_musa_silu_and_mul.py
|
||||
pytest python/sglang/multimodal_gen/test/unit/musa/layers/test_musa_rmsnorm.py
|
||||
pytest python/sglang/multimodal_gen/test/unit/musa/layers/test_musa_silu_and_mul.py
|
||||
|
||||
# =============================================== sgl-kernel ====================================================
|
||||
sgl-kernel-unit-test-musa:
|
||||
|
||||
Reference in New Issue
Block a user