Skip to content

Commit 657feba

Browse files
frost-intelfrost-intel
authored andcommitted
Nondeterministic roi_align for xpu
1 parent 7940146 commit 657feba

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/ops/roi_align.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,7 @@ def roi_align(
250250
rois = convert_boxes_to_roi_format(rois)
251251
if not torch.jit.is_scripting():
252252
if (
253-
not _has_ops() or (torch.are_deterministic_algorithms_enabled() and (input.is_cuda or input.is_mps))
253+
not _has_ops() or (torch.are_deterministic_algorithms_enabled() and (input.is_cuda or input.is_mps or input.is_xpu))
254254
) and is_compile_supported(input.device.type):
255255
return _roi_align(input, rois, spatial_scale, output_size[0], output_size[1], sampling_ratio, aligned)
256256
_assert_has_ops()

0 commit comments

Comments
 (0)