Commit 29983cc
align deterministic models list for xpu (#169640)
Summary:
Fixes intel/torch-xpu-ops#1863
In this PR, we align the deterministic models list with cpu/cuda's for xpu devices.
For your information, models like detectron2_fasterrcnn_r_50_fpn and detectron2_maskrcnn_r_50_c4 will encounter the OOM issue if using the deterministic roi_align implementation in https://github.com/pytorch/vision/blob/main/torchvision/ops/roi_align.py#L115.
X-link: pytorch/pytorch#169640
Approved by: https://github.com/jianyizh, https://github.com/EikanWang, https://github.com/guangyey, https://github.com/desertfire
Reviewed By: wdvr
Differential Revision: D91862770
fbshipit-source-id: e43188fabb61fe6ab1d3293d4fcdf3cd82cd240f1 parent eda31bc commit 29983cc
1 file changed
+1
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
3829 | 3829 | | |
3830 | 3830 | | |
3831 | 3831 | | |
3832 | | - | |
| 3832 | + | |
3833 | 3833 | | |
3834 | 3834 | | |
3835 | 3835 | | |
| |||
0 commit comments