1import torch 2 3 4__all__ = [ 5 "get_cpu_capability", 6] 7 8 9def get_cpu_capability() -> str: 10 r"""Return cpu capability as a string value. 11 12 Possible values: 13 - "DEFAULT" 14 - "VSX" 15 - "Z VECTOR" 16 - "NO AVX" 17 - "AVX2" 18 - "AVX512" 19 """ 20 return torch._C._get_cpu_capability() 21