torch.backends.cpu 的源代码

import torch

__all__ = [
    "get_cpu_capability",
]


def get_cpu_capability() -> str:
    r"""返回cpu能力作为一个字符串值。

    可能的值：
    - "DEFAULT"
    - "VSX"
    - "Z VECTOR"
    - "NO AVX"
    - "AVX2"
    - "AVX512"
    """
    return torch._C._get_cpu_capability()