[xpu][test] Port 2 test/quantization_{qat, quant_api} UT files to intel XPU #3351
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3351
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 079d4b3 with merge base 1b80794 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
meta-cla
bot
added
the
CLA Signed
|
cc @liangan1 |
| # TODO: put this in a common test utils file | ||
| _CUDA_IS_AVAILABLE = torch.cuda.is_available() | ||
| _GPU_IS_AVAILABLE = torch.accelerator.is_available() | ||
| _DEVICE = get_current_accelerator_device() |
There was a problem hiding this comment.
Directly use "_DEVICE is None" in the following changes? No need to define _GPU_IS_AVAILABLE.
liangan1
added
xpu
|
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. |
|
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. |
3 participants
For #2917, This PR is targeted to port test/quantization/test_quant_api.py and test/quantization/test_qat.py to intel XPU.