mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
469383755f
Previous full PR https://github.com/pytorch/pytorch/pull/115248 is failed to merge due to fb_code is hard to debug. I also tried to submit them as two pieces, https://github.com/pytorch/pytorch/pull/118514 https://github.com/pytorch/pytorch/pull/118515. And they have passed PreCI at that time. Now I tried to split https://github.com/pytorch/pytorch/pull/115248 into smaller piece, and it is the first step of RFC https://github.com/pytorch/pytorch/issues/124245. Changes: 1. Add cpp builder code, the new cpp_builder support Windows OS. 2. Add CPU ISA checker which is cross OS and exported from backend cpuinfo. 3. Switch compiler ISA checker to new cpp builder. 4. CppCodeCache use the new ISA checker. 5. Add temprary `test_new_cpp_build_logical` UT to help on transfer to new code. <img width="1853" alt="Image" src="https://github.com/pytorch/pytorch/assets/8433590/ce6519ab-ba92-4204-b1d6-7d15d2ba2cbe"> Pull Request resolved: https://github.com/pytorch/pytorch/pull/124045 Approved by: https://github.com/jgong5, https://github.com/jansel
17 lines
501 B
C++
17 lines
501 B
C++
#include <ATen/cpu/Utils.h>
|
|
#include <torch/csrc/cpu/Module.h>
|
|
#include <torch/csrc/utils/pybind.h>
|
|
|
|
namespace torch::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_avx2", at::cpu::is_cpu_support_avx2);
|
|
cpu.def("_is_cpu_support_avx512", at::cpu::is_cpu_support_avx512);
|
|
cpu.def("_is_cpu_support_vnni", at::cpu::is_cpu_support_vnni);
|
|
}
|
|
|
|
} // namespace torch::cpu
|