[MUSA][23/N] CI: Fix torchada preflight lock cleanup and add LLM server smoke test (#27242)
Co-authored-by: xinyue.fan <xinyue.fan@mthreads.com>
This commit is contained in:
+11
-1
@@ -19,6 +19,7 @@ HW_MAPPING = {
|
||||
"cpu": HWBackend.CPU,
|
||||
"cuda": HWBackend.CUDA,
|
||||
"amd": HWBackend.AMD,
|
||||
"musa": HWBackend.MUSA,
|
||||
"npu": HWBackend.NPU,
|
||||
"xpu": HWBackend.XPU,
|
||||
}
|
||||
@@ -42,6 +43,7 @@ PER_COMMIT_SUITES = {
|
||||
"stage-c-test-large-8-gpu-amd",
|
||||
"stage-c-test-large-8-gpu-amd-mi35x",
|
||||
],
|
||||
HWBackend.MUSA: [],
|
||||
HWBackend.CUDA: [
|
||||
"base-a-test-1-gpu-small",
|
||||
"base-b-test-1-gpu-small",
|
||||
@@ -122,6 +124,9 @@ NIGHTLY_SUITES = {
|
||||
# MI35x 8-GPU suite (different model configs)
|
||||
"nightly-amd-8-gpu-mi35x",
|
||||
],
|
||||
HWBackend.MUSA: [
|
||||
"nightly-musa-1-gpu",
|
||||
],
|
||||
HWBackend.CPU: [],
|
||||
HWBackend.NPU: [
|
||||
"nightly-1-npu-a3",
|
||||
@@ -150,7 +155,12 @@ OTHER_SUITES = {
|
||||
}
|
||||
|
||||
|
||||
_SUITE_CHECKED_BACKENDS = {HWBackend.CUDA, HWBackend.CPU, HWBackend.XPU}
|
||||
_SUITE_CHECKED_BACKENDS = {
|
||||
HWBackend.CUDA,
|
||||
HWBackend.CPU,
|
||||
HWBackend.MUSA,
|
||||
HWBackend.XPU,
|
||||
}
|
||||
|
||||
|
||||
def _valid_suites_by_backend() -> dict:
|
||||
|
||||
Reference in New Issue
Block a user