minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[AMD] CI - add partitions for stage-b-test-small-1-gpu-amd (#17345)

Browse Source
This commit is contained in:
YC Tseng
2026-01-19 08:16:07 -08:00
committed by GitHub
parent 2ea02f0642
commit 1a053a810c
5 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test-amd.yml
+2 -2
View File
@@ -211,7 +211,7 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
runner: [linux-mi325-gpu-1] runner: [linux-mi325-gpu-1]
part: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11] part: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
runs-on: ${{matrix.runner}} runs-on: ${{matrix.runner}}
steps: steps:
- name: Checkout code - name: Checkout code
@@ -233,7 +233,7 @@ jobs:
- name: Run test - name: Run test
timeout-minutes: 30 timeout-minutes: 30
run: | run: |
bash scripts/ci/amd_ci_exec.sh -w "/sglang-checkout/test" python3 run_suite.py --hw amd --suite stage-b-test-small-1-gpu-amd --auto-partition-id ${{ matrix.part }} --auto-partition-size 12 --timeout-per-file 1800 bash scripts/ci/amd_ci_exec.sh -w "/sglang-checkout/test" python3 run_suite.py --hw amd --suite stage-b-test-small-1-gpu-amd --auto-partition-id ${{ matrix.part }} --auto-partition-size 13 --timeout-per-file 1800
stage-b-test-small-1-gpu-amd-mi35x: stage-b-test-small-1-gpu-amd-mi35x:
needs: [check-changes, stage-a-test-1-amd] needs: [check-changes, stage-a-test-1-amd]
test/registered/amd/test_deepseek_v32_basic.py
+2 -2
View File
@@ -32,7 +32,7 @@ class TestDeepseekV32DP(CustomTestCase):
"8", "8",
"--enable-dp-attention", "--enable-dp-attention",
"--model-loader-extra-config", "--model-loader-extra-config",
'{"enable_multithread_load": true, "num_threads": 64}', '{"enable_multithread_load": true}',
] ]
if is_in_amd_ci(): if is_in_amd_ci():
other_args += [ other_args += [
@@ -101,7 +101,7 @@ class TestDeepseekV32TP(CustomTestCase):
"--tp", "--tp",
"8", "8",
"--model-loader-extra-config", "--model-loader-extra-config",
'{"enable_multithread_load": true, "num_threads": 64}', '{"enable_multithread_load": true}',
] ]
if is_in_amd_ci(): if is_in_amd_ci():
other_args += [ other_args += [
test/registered/amd/test_deepseek_v32_mtp.py
+2 -2
View File
@@ -44,7 +44,7 @@ class TestDeepseekV32DPMTP(CustomTestCase):
"--mem-frac", "--mem-frac",
"0.7", "0.7",
"--model-loader-extra-config", "--model-loader-extra-config",
'{"enable_multithread_load": true, "num_threads": 64}', '{"enable_multithread_load": true}',
] ]
if is_in_amd_ci(): if is_in_amd_ci():
other_args += [ other_args += [
@@ -137,7 +137,7 @@ class TestDeepseekV32TPMTP(CustomTestCase):
"--mem-frac", "--mem-frac",
"0.7", "0.7",
"--model-loader-extra-config", "--model-loader-extra-config",
'{"enable_multithread_load": true, "num_threads": 64}', '{"enable_multithread_load": true}',
] ]
if is_in_amd_ci(): if is_in_amd_ci():
other_args += [ other_args += [
test/registered/attention/test_triton_sliding_window.py
+1 -1
View File
@@ -16,7 +16,7 @@ from sglang.test.test_utils import (
# Sliding window attention with Triton backend (Gemma-3 model) # Sliding window attention with Triton backend (Gemma-3 model)
register_cuda_ci(est_time=100, suite="stage-b-test-large-1-gpu") register_cuda_ci(est_time=100, suite="stage-b-test-large-1-gpu")
register_amd_ci(est_time=100, suite="stage-b-test-small-1-gpu-amd") register_amd_ci(est_time=200, suite="stage-b-test-small-1-gpu-amd")
class TestSlidingWindowAttentionTriton(CustomTestCase): class TestSlidingWindowAttentionTriton(CustomTestCase):
test/registered/openai_server/basic/test_openai_server.py
+1 -1
View File
@@ -29,7 +29,7 @@ from sglang.test.test_utils import (
) )
register_cuda_ci(est_time=184, suite="stage-b-test-small-1-gpu") register_cuda_ci(est_time=184, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=149, suite="stage-b-test-small-1-gpu-amd") register_amd_ci(est_time=200, suite="stage-b-test-small-1-gpu-amd")
class TestOpenAIServer(CustomTestCase): class TestOpenAIServer(CustomTestCase):