RYDE-WORK/ktransformers
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/ktransformers.git synced 2026-02-06 14:39:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix merge bug, this branch also padding Marlin

Browse Source
This commit is contained in:
Atream 2025-02-22 09:00:09 +00:00
parent 024009675e
commit f7f1059873
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ktransformers/ktransformers_ext/cuda/binding.cpp
View File

@ -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("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("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")); py::arg("size_n"), py::arg("size_k"), py::arg("is_k_full"));
#endif
} }