mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
e56dcf2772
This PR removes `torch.cpu._is_arm_sve_supported()` and replaces is with stable `torch.backends.cpu.get_cpu_capability()` I should have reviewed https://github.com/pytorch/pytorch/pull/134672 more thoroughly, because it introduced duplicate, but slightly different API for detecting CPU architectures, which resulted in runtime crashes on system that do support SVE128, rather than SVE256 Fixes https://github.com/pytorch/pytorch/issues/145441 Pull Request resolved: https://github.com/pytorch/pytorch/pull/146207 Approved by: https://github.com/angelayi
14 lines
434 B
Python
14 lines
434 B
Python
from torch.types import _bool, _int
|
|
|
|
# Defined in torch/csrc/cpu/Module.cpp
|
|
|
|
def _is_avx2_supported() -> _bool: ...
|
|
def _is_avx512_supported() -> _bool: ...
|
|
def _is_avx512_vnni_supported() -> _bool: ...
|
|
def _is_avx512_bf16_supported() -> _bool: ...
|
|
def _is_amx_tile_supported() -> _bool: ...
|
|
def _is_amx_fp16_supported() -> _bool: ...
|
|
def _init_amx() -> _bool: ...
|
|
def _L1d_cache_size() -> _int: ...
|
|
def _L2_cache_size() -> _int: ...
|