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"
    """
    return torch._C._get_cpu_capability()