[AMD] Add TP=8 models to nightly test and make TP=2 test stable (#15296)

This commit is contained in:
michael-amd
2025-12-19 13:15:19 -08:00
committed by GitHub
parent bd16244d93
commit 2ee6c810b8
5 changed files with 1128 additions and 30 deletions
+4
View File
@@ -317,6 +317,10 @@ suite_amd = {
"nightly-amd": [
TestFile("nightly/test_gsm8k_eval_amd.py"),
],
# AMD 8-GPU tests for base models using gsm8k completion benchmark
"nightly-amd-8-gpu": [
TestFile("nightly/test_gsm8k_completion_eval_amd.py"),
],
}
# Add Intel Xeon tests