diff --git a/python/src/ops.cpp b/python/src/ops.cpp index 5ea979670..b58ee670f 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -1460,7 +1460,7 @@ void init_ops(nb::module_& m) { )pbdoc"); m.def( "kron", - &kron, + &mx::kron, nb::arg("a"), nb::arg("b"), nb::kw_only(),