mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
7614338b69
This reverts commit 92284fb2ff.
Reverted https://github.com/pytorch/pytorch/pull/158932 on behalf of https://github.com/malfet due to Hmm, but from OSS point of view, this is a no-op ([comment](https://github.com/pytorch/pytorch/pull/158932#issuecomment-3387961238))
22 lines
314 B
Python
22 lines
314 B
Python
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()
|