🐛 Describe the bug

Please see this line on the main branch.

It decides whether the image resize can be done for uint8 datatype with native CPU.

However, the return type of torch.backends.cpu.get_cpu_capability() is not a kind of set or sequence of strs. It returns str, https://pytorch.org/docs/2.2/backends.html#torch.backends.cpu.get_cpu_capability. Therefore, on a AVX512 support machine, torch.backends.cpu.get_cpu_capability() returns "AVX512" and it leads to float32 casting at

although this machine can support AVX2.

Versions

torch>=2.1 and torchvision>=0.16.1