[Intel XPU] Add common code for UTs on XPU #3335
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3335
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit f4eed65 with merge base ff0e461 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
meta-cla
bot
added
the
CLA Signed
| pytest -v -s torchao/test/quantization/quantize_/workflows/int4/test_int4_plain_int32_tensor.py | ||
| pytest -v -s torchao/test/quantization/quantize_/workflows/int4/test_int4_plain_int32_tensor.py | ||
|
|
||
| pytest -v -s torchao/test/quantization/ |
There was a problem hiding this comment.
does these tests run on xpu?
There was a problem hiding this comment.
These UTs will be run on XPU finally. The current TorchAO CI should only cover the CUDA device and skip on XPU. We plan to firstly add these test folders into the XPU ci which is triggered by ciflow/xpu and then enabled the XPU UTs file by file to reduce the review effort.
liangan1
added
topic: for developers
|
To add the ciflow label This helps ensure we don't trigger CI on this PR until it is actually authorized to do so. Please ping one of the reviewers if you do not have access to approve and run workflows. |
torchao/testing/utils.py
Outdated
| return decorator | ||
|
|
||
|
|
||
| def skip_if_no_xpu(message=None): |
There was a problem hiding this comment.
currently these things are a bit confusing I think, since I remember some of these work with unittests and others work with pytest. can you confirm what these are working with? I think ideally we should create two versions, one for unittests one for pytest
There was a problem hiding this comment.
Created two versions for both unittest and pytest
zxd1997066
force-pushed
the
xiangdong/xpu_ut
branch
from
d828aec to
b4933ae
Compare
zxd1997066
force-pushed
the
xiangdong/xpu_ut
branch
from
318c66b to
f4eed65
Compare
3 participants
This PR is targeted to add common code for UTs on XPU