Function PYBIND11_MODULE

quantizer.cpp:68–71

}

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.def("quantize_q2_k", &quantize_q2_k, "Quantize a tensor to Q2_K format");
  m.def("quantize_q3_k", &quantize_q3_k, "Quantize a tensor to Q3_K format");
}