xref: /aosp_15_r20/external/pytorch/torch/backends/cpu/__init__.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1import torch
2
3
4__all__ = [
5    "get_cpu_capability",
6]
7
8
9def get_cpu_capability() -> str:
10    r"""Return cpu capability as a string value.
11
12    Possible values:
13    - "DEFAULT"
14    - "VSX"
15    - "Z VECTOR"
16    - "NO AVX"
17    - "AVX2"
18    - "AVX512"
19    """
20    return torch._C._get_cpu_capability()
21