torch.backends.cpu — PyTorch 2.7 documentation (original) (raw)

Source code for torch.backends.cpu

import torch

all = [ "get_cpu_capability", ]

[docs]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()