1 #include <torch/extension.h> 2 foo()3 void foo() { } 4 TORCH_LIBRARY_IMPL(__test,CPU,m)5 TORCH_LIBRARY_IMPL(__test, CPU, m) { 6 m.impl("foo", foo); 7 } 8 PYBIND11_MODULE(TORCH_EXTENSION_NAME,m)9 PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { 10 m.def("bar", foo); 11 } 12