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

fix test (#627)

Browse Source
This commit is contained in:
Awni Hannun
2024-02-04 16:18:03 -08:00
committed by GitHub
parent ebfd3618b0
commit 5c3ac52dd7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/src/array.cpp
View File

@@ -1110,7 +1110,8 @@ void init_array(py::module_& m) {
py::kw_only(), py::kw_only(),
"stream"_a = none, "stream"_a = none,
"See :func:`abs`.") "See :func:`abs`.")
.def("__abs__", &mlx::core::abs, "See :func:`abs`.") .def(
"__abs__", [](const array& a) { return abs(a); }, "See :func:`abs`.")
.def( .def(
"square", "square",
&square, &square,