File size: 205 Bytes
3224250 |
1 2 3 4 5 6 7 8 9 10 11 12 |
#include "grouped_gemm.h"
#include <torch/extension.h>
namespace grouped_gemm {
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("gmm", &GroupedGemm, "Grouped GEMM.");
}
} // namespace grouped_gemm
|