Unverified Commit 5326c898 authored by Kunshang Ji's avatar Kunshang Ji Committed by GitHub
Browse files

[XPU][CI]skip test_preprocess_error_handling due to fork/spawn issue (#31381)


Signed-off-by: default avatarKunshang Ji <kunshang.ji@intel.com>
parent 87f1b8ca
......@@ -15,9 +15,10 @@ MODEL_NAME = "hmellor/tiny-random-LlamaForCausalLM"
def test_preprocess_error_handling(monkeypatch: pytest.MonkeyPatch):
"""Test that preprocessing errors are handled gracefully."""
if current_platform.is_rocm():
if current_platform.is_rocm() or current_platform.is_xpu():
pytest.skip(
"Skipped on ROCm: this test only works with 'fork', but ROCm uses 'spawn'."
"Skipped on ROCm/XPU: this test only works with 'fork', "
"but ROCm/XPU uses 'spawn'."
)
assert not torch.cuda.is_initialized(), (
......
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