pytorch/torch/_C/_cpu.pyi
leslie-fang-intel 9832cfbbfe Quantization oneDNN backend only support VNNI CPU (#103653)
**Summary**

- Update the quantization document that default qconfig with oneDNN backend is recommended to be used on CPUs with Vector Neural Network Instruction support.
- Add the warning message when user uses default qconfig with oneDNN backend on CPU without Vector Neural Network Instruction support.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/103653
Approved by: https://github.com/jgong5, https://github.com/malfet
2023-06-19 09:50:07 +00:00

6 lines
112 B
Python

from torch.types import _bool
# Defined in torch/csrc/cpu/Module.cpp
def _is_cpu_support_vnni() -> _bool: ...