#pragma once #include namespace torch { namespace cpu { void initModule(PyObject* module); } // namespace cpu } // namespace torch