[xpu][test] Port 2 test/dtypes_{floatx, bitpacking} UT files to intel XPU #3368
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3368
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit b6a2c80 with merge base 257d18a ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
meta-cla
bot
added
the
CLA Signed
|
cc @liangan1 |
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. |
liangan1
added
ciflow/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. |
test/dtypes/test_floatx.py
Outdated
| _DEVICES = ( | ||
| ["cpu"] | ||
| + (["cuda"] if torch.cuda.is_available() else []) | ||
| + (["xpu"] if torch.xpu.is_available() else []) | ||
| ) |
There was a problem hiding this comment.
why not do torch.accelerator.is_available here (and change cuda to get_current_accelerator_device)?
There was a problem hiding this comment.
updated to use torch.accelerator.is_available
… XPU (pytorch#3368) * enable test/dtypes/test_bitpacking.py on intel xpu * enable test/dtypes/test_floatx.py * enable test/dtypes/test_floatx.py * fix format issue * fix format issue * update _DEVICES
3 participants
For #2917, This PR is targeted to port test/dtypes/test_floatx.py and test/dtypes/test_bitpacking.py to intel XPU.