ci: fix xpu skip condition for test_model_parallel_beam_search #35742
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
return unittest.skip()used in thetest_model_parallel_beam_searchin skip condition for xpu did not actually mark test to be skipped running under pytest:Other tests use
self.skipTest(). Reusing this approach and moving the condition outside the loop (since it does not depend on it) allows to skip for xpu correctly:Secondly,
device_map="auto"is now implemented for XPU for IPEX>=2.5 and torch>=2.6, so we can now enable these tests for XPU for new IPEX/torch versions.Fixes: 1ea3ad1 ("[tests] use
torch_deviceinstead ofautofor model testing (#29531)")CC: @faaany @ydshieh