benchmark
29983cca - align deterministic models list for xpu (#169640)

Commit
59 days ago
align deterministic models list for xpu (#169640) Summary: Fixes https://github.com/intel/torch-xpu-ops/issues/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: https://github.com/pytorch/pytorch/pull/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: e43188fabb61fe6ab1d3293d4fcdf3cd82cd240f
Author
generatedunixname499836121
Committer
Parents
Loading