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

[AMD] Migrate 2-GPU kernel allreduce tests into the registered system (#27722)

Browse Source 
Co-authored-by: michaelzhang-ai <michaelzhang@example.com>
This commit is contained in:
Michael
2026-06-09 19:39:03 -07:00
committed by GitHub
co-authored by michaelzhang-ai
parent 7e3e616159
commit f42a093261
7 changed files with 14 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test-amd-rocm720.yml
+1 -2
View File
@@ -272,8 +272,7 @@ jobs:
- name: Run test
timeout-minutes: 20
run: |
docker exec -w /sglang-checkout/sgl-kernel/tests ci_sglang python3 -m pytest test_amd_deterministic_custom_allreduce.py
docker exec -w /sglang-checkout/sgl-kernel/tests ci_sglang python3 -m pytest test_amd_nccl_allreduce_determinism.py
bash scripts/ci/amd/amd_ci_exec.sh -w "/sglang-checkout/test" python3 run_suite.py --hw amd --suite sgl-kernel-unit-test-2-gpu-amd
# =============================================== primary ====================================================
.github/workflows/pr-test-amd.yml
+1 -11
View File
@@ -283,17 +283,7 @@ jobs:
env:
CONTINUE_ON_ERROR: ${{ needs.check-changes.outputs.continue_on_error }}
run: |
failures=0
run_pytest() {
if [[ "$CONTINUE_ON_ERROR" == "true" ]]; then
"$@" || failures=$((failures + 1))
else
"$@"
fi
}
run_pytest docker exec -w /sglang-checkout/sgl-kernel/tests ci_sglang python3 -m pytest test_amd_deterministic_custom_allreduce.py
run_pytest docker exec -w /sglang-checkout/sgl-kernel/tests ci_sglang python3 -m pytest test_amd_nccl_allreduce_determinism.py
exit $failures
bash scripts/ci/amd/amd_ci_exec.sh -w "/sglang-checkout/test" python3 run_suite.py --hw amd --suite sgl-kernel-unit-test-2-gpu-amd ${{ needs.check-changes.outputs.continue_on_error == 'true' && '--continue-on-error' || '' }}
# =============================================== primary ====================================================