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 00:59:28 -08:00
committed by GitHub
co-authored by Sai Enduri
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
.github/workflows/pr-test-amd.yml
+3 -5
View File
@@ -458,24 +458,22 @@ jobs:
unit-test-backend-8-gpu-amd: unit-test-backend-8-gpu-amd:
needs: [check-changes, unit-test-backend-2-gpu-amd] needs: [check-changes, unit-test-backend-2-gpu-amd]
# Temporarily disabled - uncomment when ready to re-enable
if: | if: |
always() && always() &&
( (
(inputs.target_stage == 'unit-test-backend-8-gpu-amd') || (inputs.target_stage == 'unit-test-backend-8-gpu-amd') ||
( (
false &&
!inputs.target_stage && !inputs.target_stage &&
(!failure() && !cancelled()) && (!failure() && !cancelled()) &&
((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true')) ((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true'))
) )
) )
env: env:
RUNNER_LABELS: linux-mi300-gpu-8 RUNNER_LABELS: linux-mi325-gpu-8
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
runner: [linux-mi300-gpu-8] runner: [linux-mi325-gpu-8]
part: [0, 1] part: [0, 1]
runs-on: ${{matrix.runner}} runs-on: ${{matrix.runner}}
steps: steps:
@@ -751,7 +749,7 @@ jobs:
stage-a-test-1-amd, stage-a-test-1-amd,
unit-test-backend-1-gpu-amd, unit-test-backend-1-gpu-amd,
unit-test-backend-2-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-1-amd,
performance-test-1-gpu-part-2-amd, performance-test-1-gpu-part-2-amd,
performance-test-2-gpu-amd, performance-test-2-gpu-amd,
test/srt/test_deepseek_v3_mtp.py
+4 -1
View File
@@ -80,7 +80,10 @@ class TestDeepseekV3MTP(CustomTestCase):
f"{avg_spec_accept_length=:.2f}\n" f"{avg_spec_accept_length=:.2f}\n"
) )
self.assertGreater(metrics["accuracy"], 0.935) 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): def test_bs_1_speed(self):
args = BenchArgs(port=int(self.base_url.split(":")[-1]), max_new_tokens=2048) args = BenchArgs(port=int(self.base_url.split(":")[-1]), max_new_tokens=2048)