[XPU][CI] increase timeout of extract_hidden_states tp2 (#53862)
liuzhenwei committed
Aug 27, 2026 at 06:57 UTC
a27e88c8627bcdd86f1cd35e0c84830944ad5b3c
1 file changed
+3
-1
tests/v1/kv_connector/extract_hidden_states_integration/test_extraction.py
+3
-1
@@ -296,7 +296,9 @@ def test_extract_hidden_states_qwen35_hybrid_smoke(tmp_path):
296
)
297
298
299
-@pytest.mark.timeout(240 if current_platform.is_rocm() else 120)
299
+@pytest.mark.timeout(
300
+ 240 if current_platform.is_rocm() or current_platform.is_xpu() else 120
301
+)
302
@multi_gpu_test(num_gpus=2)
303
@create_new_process_for_each_test()
304
def test_extract_hidden_states_tp2():