blob: 353ad12f5b7d3eb76296a7e80780a488675fb43a [file] [log] [blame]
from torch.types import _bool, _int
# Defined in torch/csrc/cpu/Module.cpp
def _is_cpu_support_avx2() -> _bool: ...
def _is_cpu_support_avx512() -> _bool: ...
def _is_cpu_support_avx512_vnni() -> _bool: ...
def _is_cpu_support_amx_tile() -> _bool: ...
def _init_amx() -> _bool: ...
def _L1d_cache_size() -> _int: ...
def _L2_cache_size() -> _int: ...