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

More jitting (#1132)

Browse Source 
* docs + circle min size build

* jit scan, arange, softmax

* add sort

* jit reductions

* remove print

* fix deps

* clean includes / nits
This commit is contained in:
Awni Hannun
2024-05-23 16:23:44 -07:00
committed by GitHub
parent 9401507336
commit 0189ab6ab6
41 changed files with 2377 additions and 1846 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mlx/primitives.h
View File

@@ -1565,8 +1565,9 @@ class Reduce : public UnaryPrimitive {
switch (reduce_type_) {
case And:
os << "And";
break;
case Or:
os << "And";
os << "Or";
break;
case Sum:
os << "Sum";
@@ -1581,7 +1582,6 @@ class Reduce : public UnaryPrimitive {
os << "Max";
break;
}
os << " Reduce";
}
bool is_equivalent(const Primitive& other) const override;
@@ -1647,7 +1647,6 @@ class Scan : public UnaryPrimitive {
os << "Max";
break;
}
os << " Reduce";
}
bool is_equivalent(const Primitive& other) const override;