From f7f10598734984ac6fe62fd8821bb01f8c4f0599 Mon Sep 17 00:00:00 2001 From: Atream Date: Sat, 22 Feb 2025 09:00:09 +0000 Subject: [PATCH] fix merge bug, this branch also padding Marlin --- ktransformers/ktransformers_ext/cuda/binding.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/ktransformers/ktransformers_ext/cuda/binding.cpp b/ktransformers/ktransformers_ext/cuda/binding.cpp index 75cfcdb..0b1994d 100644 --- a/ktransformers/ktransformers_ext/cuda/binding.cpp +++ b/ktransformers/ktransformers_ext/cuda/binding.cpp @@ -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 }