mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
9832cfbbfe
**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
17 lines
392 B
C++
17 lines
392 B
C++
#include <ATen/cpu/Utils.h>
|
|
#include <torch/csrc/cpu/Module.h>
|
|
#include <torch/csrc/utils/pybind.h>
|
|
|
|
namespace torch {
|
|
namespace cpu {
|
|
|
|
void initModule(PyObject* module) {
|
|
auto m = py::handle(module).cast<py::module>();
|
|
|
|
auto cpu = m.def_submodule("_cpu", "cpu related pybind.");
|
|
cpu.def("_is_cpu_support_vnni", at::cpu::is_cpu_support_vnni);
|
|
}
|
|
|
|
} // namespace cpu
|
|
} // namespace torch
|