Skip to content

Commit 29983cc

Browse files
generatedunixname499836121meta-codesync[bot]
authored andcommitted
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: e43188fabb61fe6ab1d3293d4fcdf3cd82cd240f
1 parent eda31bc commit 29983cc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

userbenchmark/dynamo/dynamobench/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3829,7 +3829,7 @@ def setup_determinism_for_accuracy_test(args):
38293829
# some of the models do not support use_deterministic_algorithms
38303830
torch.use_deterministic_algorithms(True)
38313831

3832-
if args.devices == ["rocm"] or args.devices == ["xpu"]:
3832+
if args.devices == ["rocm"]:
38333833
torch.use_deterministic_algorithms(True, warn_only=True)
38343834

38353835
torch.backends.cudnn.deterministic = True

0 commit comments

Comments
 (0)