pytorch
6a12f10b - Publicly exposing `torch.backends.cpu.get_cpu_capability()` (#100164)

Commit
1 year ago
Publicly exposing `torch.backends.cpu.get_cpu_capability()` (#100164) Description: - As suggested by Nikita, created `torch.backends.cpu` submodule and exposed `get_cpu_capability`. - In torchvision Resize method we want to know current cpu capability in order to pick appropriate codepath depending on cpu capablities Newly coded vectorized resize of uint8 images on AVX2 supported CPUs is now faster than older way (uint8->float->resize->uint8). However, on non-avx hardware (e.g. Mac M1) certain configs are slower using native uint8. Pull Request resolved: https://github.com/pytorch/pytorch/pull/100164 Approved by: https://github.com/albanD, https://github.com/malfet
Author
Committer
Parents
Loading