drbh

feat: vendor grouped gemm

3224250 over 1 year ago

205 Bytes

#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

	#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