[xpu][test] Port 2 test/quantization_{qat, quant_api} UT files to intel XPU #3351
[xpu][test] Port 2 test/quantization_{qat, quant_api} UT files to intel XPU #3351liangan1 merged 4 commits intopytorch:mainfrom
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. |
|
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.
|
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. |
…el XPU (pytorch#3351) * port 2 files to intel XPU * port 2 files to intel XPU * update * update
For #2917, This PR is targeted to port test/quantization/test_quant_api.py and test/quantization/test_qat.py to intel XPU.