chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

[AMD] add unit-test-backend-8-gpu-amd back (#15253)

Browse Source 
Co-authored-by: Sai Enduri <saimanas.enduri@amd.com>
This commit is contained in:
yctseng0211
2025-12-19 16:59:28 +08:00
committed by GitHub
parent a21aa87ec5
commit af780c59f3
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
.github/workflows/pr-test-amd.yml vendored
View File

@@ -458,24 +458,22 @@ jobs:
unit-test-backend-8-gpu-amd:
needs: [check-changes, unit-test-backend-2-gpu-amd]
# Temporarily disabled - uncomment when ready to re-enable
if: |
always() &&
(
(inputs.target_stage == 'unit-test-backend-8-gpu-amd') ||
(
false &&
!inputs.target_stage &&
(!failure() && !cancelled()) &&
((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true'))
)
)
env:
RUNNER_LABELS: linux-mi300-gpu-8
RUNNER_LABELS: linux-mi325-gpu-8
strategy:
fail-fast: false
matrix:
runner: [linux-mi300-gpu-8]
runner: [linux-mi325-gpu-8]
part: [0, 1]
runs-on: ${{matrix.runner}}
steps:
@@ -751,7 +749,7 @@ jobs:
stage-a-test-1-amd,
unit-test-backend-1-gpu-amd,
unit-test-backend-2-gpu-amd,
# unit-test-backend-8-gpu-amd, # Temporarily disabled
unit-test-backend-8-gpu-amd,
performance-test-1-gpu-part-1-amd,
performance-test-1-gpu-part-2-amd,
performance-test-2-gpu-amd,

5
test/srt/test_deepseek_v3_mtp.py
View File

@@ -80,7 +80,10 @@ class TestDeepseekV3MTP(CustomTestCase):
f"{avg_spec_accept_length=:.2f}\n"
)
self.assertGreater(metrics["accuracy"], 0.935)
self.assertGreater(avg_spec_accept_length, 2.9)
if is_in_amd_ci():
self.assertGreater(avg_spec_accept_length, 2.8)
else:
self.assertGreater(avg_spec_accept_length, 2.9)
def test_bs_1_speed(self):
args = BenchArgs(port=int(self.base_url.split(":")[-1]), max_new_tokens=2048)