[XPU] Add registry mechanism for XPU CI tests (#25405)
Co-authored-by: Ma Mingfei <mingfei.ma@intel.com>
This commit is contained in:
co-authored by
Ma Mingfei
parent
87c3171aaa
commit
737c6cd6d1
+7
-1
@@ -20,6 +20,7 @@ HW_MAPPING = {
|
||||
"cuda": HWBackend.CUDA,
|
||||
"amd": HWBackend.AMD,
|
||||
"npu": HWBackend.NPU,
|
||||
"xpu": HWBackend.XPU,
|
||||
}
|
||||
|
||||
# Per-commit test suites (run on every PR).
|
||||
@@ -77,6 +78,10 @@ PER_COMMIT_SUITES = {
|
||||
"stage-b-test-4-npu-a3",
|
||||
"stage-b-test-16-npu-a3",
|
||||
],
|
||||
HWBackend.XPU: [
|
||||
"stage-a-test-1-gpu-xpu",
|
||||
"stage-b-test-1-gpu-xpu",
|
||||
],
|
||||
}
|
||||
|
||||
# Nightly test suites (run nightly, organized by GPU configuration)
|
||||
@@ -127,6 +132,7 @@ NIGHTLY_SUITES = {
|
||||
"full-8-npu-a3",
|
||||
"full-16-npu-a3",
|
||||
],
|
||||
HWBackend.XPU: [],
|
||||
}
|
||||
|
||||
|
||||
@@ -141,7 +147,7 @@ OTHER_SUITES = {
|
||||
}
|
||||
|
||||
|
||||
_SUITE_CHECKED_BACKENDS = {HWBackend.CUDA, HWBackend.CPU}
|
||||
_SUITE_CHECKED_BACKENDS = {HWBackend.CUDA, HWBackend.CPU, HWBackend.XPU}
|
||||
|
||||
|
||||
def _valid_suites_by_backend() -> dict:
|
||||
|
||||
Reference in New Issue
Block a user