_cpu.pyi 434 B

12345678910111213
  1. from torch.types import _bool, _int
  2. # Defined in torch/csrc/cpu/Module.cpp
  3. def _is_avx2_supported() -> _bool: ...
  4. def _is_avx512_supported() -> _bool: ...
  5. def _is_avx512_vnni_supported() -> _bool: ...
  6. def _is_avx512_bf16_supported() -> _bool: ...
  7. def _is_amx_tile_supported() -> _bool: ...
  8. def _is_amx_fp16_supported() -> _bool: ...
  9. def _init_amx() -> _bool: ...
  10. def _L1d_cache_size() -> _int: ...
  11. def _L2_cache_size() -> _int: ...