mirror of
https://github.com/RYDE-WORK/ktransformers.git
synced 2026-01-28 01:43:37 +08:00
fix merge bug, this branch also padding Marlin
This commit is contained in:
parent
024009675e
commit
f7f1059873
@ -60,4 +60,5 @@ PYBIND11_MODULE(KTransformersOps, m) {
|
||||
py::arg("a"), py::arg("b_q_weight"), py::arg("b_scales"), py::arg("g_idx"),
|
||||
py::arg("perm"), py::arg("workspace"), py::arg("num_bits"), py::arg("size_m"),
|
||||
py::arg("size_n"), py::arg("size_k"), py::arg("is_k_full"));
|
||||
#endif
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user