zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-07 04:38:13 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix the optional in gather_qmm python binding

Browse Source
This commit is contained in:
Angelos Katharopoulos
2024-12-16 22:14:19 -08:00
parent 4847199ec6
commit 14420949d2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/ops.cpp
View File

@@ -4206,7 +4206,7 @@ void init_ops(nb::module_& m) {
nb::arg(), nb::arg(),
nb::arg(), nb::arg(),
"scales"_a, "scales"_a,
"biases"_a, "biases"_a = nb::none(),
"lhs_indices"_a = nb::none(), "lhs_indices"_a = nb::none(),
"rhs_indices"_a = nb::none(), "rhs_indices"_a = nb::none(),
"transpose"_a = true, "transpose"_a = true,