We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4bc406d commit 3db0b12Copy full SHA for 3db0b12
torchvision/transforms/v2/functional/_geometry.py
@@ -261,7 +261,7 @@ def _do_native_uint8_resize_on_cpu(interpolation: InterpolationMode) -> bool:
261
if torch.compiler.is_compiling():
262
return True
263
else:
264
- return "AVX2" in torch.backends.cpu.get_cpu_capability()
+ return torch.backends.cpu.get_cpu_capability() in ("AVX2", "AVX512")
265
266
return interpolation == InterpolationMode.BICUBIC
267
0 commit comments