pytorch
622e582a - Register get_cpu_capability for jit (#100723)

Commit
1 year ago
Register get_cpu_capability for jit (#100723) Description: Context: In torchvision we ensure that functional ops are torchscriptable. Recently exposed `torch.backends.cpu.get_cpu_capability()` in https://github.com/pytorch/pytorch/pull/100164 is failing in torchvision CI ``` RuntimeError: Python builtin <built-in function _get_cpu_capability> is currently not supported in Torchscript: File "/usr/local/lib/python3.10/dist-packages/torch/backends/cpu/__init__.py", line 17 - "AVX512" """ return torch._C._get_cpu_capability() ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE ``` Ref: https://github.com/pytorch/vision/pull/7557 In this PR, `torch._C._get_cpu_capability()` is explicitly registered for JIT and tested. Pull Request resolved: https://github.com/pytorch/pytorch/pull/100723 Approved by: https://github.com/albanD
Author
Committer
Parents
Loading