xref: /aosp_15_r20/external/pytorch/torch/_C/_cpu.pyi (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1from torch.types import _bool, _int
2
3# Defined in torch/csrc/cpu/Module.cpp
4
5def _is_avx2_supported() -> _bool: ...
6def _is_avx512_supported() -> _bool: ...
7def _is_avx512_vnni_supported() -> _bool: ...
8def _is_avx512_bf16_supported() -> _bool: ...
9def _is_amx_tile_supported() -> _bool: ...
10def _init_amx() -> _bool: ...
11def _L1d_cache_size() -> _int: ...
12def _L2_cache_size() -> _int: ...
13