Unverified Commit c7b2d2b3 authored by VincyZhang's avatar VincyZhang Committed by GitHub
Browse files

chore(test): introduce xpu_1 and xpu_2 pytest markers (#7414)


Signed-off-by: default avatarVincyZhang <wenxin.zhang@intel.com>
parent 50364d25
...@@ -230,6 +230,8 @@ markers = [ ...@@ -230,6 +230,8 @@ markers = [
"gpu_2: marks tests to run on 2GPUs", "gpu_2: marks tests to run on 2GPUs",
"gpu_4: marks tests to run on 4GPUs", "gpu_4: marks tests to run on 4GPUs",
"gpu_8: marks tests to run on 8GPUs", "gpu_8: marks tests to run on 8GPUs",
"xpu_1: marks tests to run on XPU",
"xpu_2: marks tests to run on 2XPUs",
"max_vram_gib(N): peak VRAM in GiB (with 10% safety). Filter with --max-vram-gib=N", "max_vram_gib(N): peak VRAM in GiB (with 10% safety). Filter with --max-vram-gib=N",
"e2e: marks tests as end-to-end tests", "e2e: marks tests as end-to-end tests",
"integration: marks tests as integration tests", "integration: marks tests as integration tests",
......
...@@ -57,7 +57,17 @@ REQUIRED_CATEGORIES: Dict[str, Set[str]] = { ...@@ -57,7 +57,17 @@ REQUIRED_CATEGORIES: Dict[str, Set[str]] = {
"multimodal", "multimodal",
"performance", "performance",
}, },
"Hardware": {"gpu_0", "gpu_1", "gpu_2", "gpu_4", "gpu_8", "h100", "k8s"}, "Hardware": {
"gpu_0",
"gpu_1",
"gpu_2",
"gpu_4",
"gpu_8",
"h100",
"k8s",
"xpu_1",
"xpu_2",
},
} }
STUB_MODULES = [ STUB_MODULES = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment