mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
3bf922a6ce
Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/106249 Approved by: https://github.com/Skylion007
20 lines
299 B
Python
20 lines
299 B
Python
import torch
|
|
|
|
__all__ = [
|
|
"get_cpu_capability",
|
|
]
|
|
|
|
|
|
def get_cpu_capability() -> str:
|
|
r"""Returns cpu capability as a string value.
|
|
|
|
Possible values:
|
|
- "DEFAULT"
|
|
- "VSX"
|
|
- "Z VECTOR"
|
|
- "NO AVX"
|
|
- "AVX2"
|
|
- "AVX512"
|
|
"""
|
|
return torch._C._get_cpu_capability()
|