#include "grouped_gemm.h" #include namespace grouped_gemm { PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { m.def("gmm", &GroupedGemm, "Grouped GEMM."); } } // namespace grouped_gemm