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