[XPU] exclude unsupported models for test_tensor_sechma.py (#47008)
Signed-off-by: Yan Ma <yan.ma@intel.com> Signed-off-by: Kunshang Ji <kunshang.ji@intel.com> Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
Yan Ma committed
Jun 29, 2026 at 20:30 UTC
59575da46df964e6161fb0e1a77fa76ea9ce3106
2 files changed
+15
-3
.buildkite/intel_jobs/models_multimodal_intel.yaml
-2
@@ -125,7 +125,5 @@ steps:
125
pip install open-clip-torch --no-deps &&
126
cd tests &&
127
pytest -v -s models/multimodal/processing/test_tensor_schema.py
128
- --deselect "tests/models/multimodal/processing/test_tensor_schema.py::test_model_tensor_schema[mistralai/Mistral-Large-3-675B-Instruct-2512-NVFP4]"
129
- --deselect "tests/models/multimodal/processing/test_tensor_schema.py::test_model_tensor_schema[Qwen/Qwen2.5-Omni-7B-AWQ]"
128
--num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB'
129
parallelism: 4
tests/models/multimodal/processing/test_common.py
+15
-1
@@ -20,6 +20,7 @@ from vllm.multimodal import MULTIMODAL_REGISTRY
20
from vllm.multimodal.cache import MultiModalProcessorOnlyCache
21
from vllm.multimodal.inputs import batched_tensors_equal
22
from vllm.multimodal.processing import BaseMultiModalProcessor, InputProcessingContext
23
+from vllm.platforms import current_platform
24
from vllm.tokenizers import TokenizerLike, cached_tokenizer_from_config
25
from vllm.utils.mistral import is_mistral_tokenizer
26
@@ -83,6 +84,12 @@ MM_DATA_PATCHES = {
84
"glmasr": glmasr_patch_mm_data,
85
}
86
87
+_XPU_EXCLUDED_MODEL_IDS = {
88
+ "baidu/Unlimited-OCR",
89
+ "mistralai/Mistral-Large-3-675B-Instruct-2512-NVFP4",
90
+ "Qwen/Qwen2.5-Omni-7B-AWQ",
91
+}
92
+
93
94
def _iter_model_ids_to_test(model_arch_list: AbstractSet[str]):
95
for model_arch in model_arch_list:
@@ -97,7 +104,14 @@ def _iter_model_ids_to_test(model_arch_list: AbstractSet[str]):
104
105
106
def _get_model_ids_to_test(model_arch_list: AbstractSet[str]):
100
- return list(_iter_model_ids_to_test(model_arch_list))
107
+ model_ids = list(_iter_model_ids_to_test(model_arch_list))
108
+
109
+ if current_platform.is_xpu():
110
+ for excluded_model_id in _XPU_EXCLUDED_MODEL_IDS:
111
+ while excluded_model_id in model_ids:
112
+ model_ids.remove(excluded_model_id)
113
+
114
+ return model_ids
115
116
117
def get_model_ids_to_test():