Add XPU to is_compile_supported to support roi_align op in torchvision (#147541)
Summary:
Part of the required fix for https://github.com/intel/torch-xpu-ops/issues/1264.
To support `roi_align`, torchvision uses `is_compile_supported` in `torch/_dynamo/utils.py` to compile a non-deterministic version of the op for backwards passes. This PR adds XPU device to the supported compile devices.
The `is_compile_supported()` util function has extremely limited usage, only being used in `torchvision.ops.roi_align` and `torch.utils._content_store.has_storage()`.
X-link: https://github.com/pytorch/pytorch/pull/147541
Approved by: https://github.com/guangyey, https://github.com/jansel
Reviewed By: jeanschmidt
Differential Revision: D70075274
fbshipit-source-id: 8cdb21eaa08243a7442383950f96c3351f5f52c5
Co-authored-by: lei,zhenyuan <zhenyuan.lei@intel.com>
Author
generatedunixname499836121