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

LogCumSumExp (#2069)

Browse Source
This commit is contained in:
Yury Popov
2025-04-13 11:27:29 +03:00
committed by GitHub
parent 7275ac7523
commit e9e268336b
15 changed files with 209 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mlx/primitives.h
View File

@@ -1728,7 +1728,7 @@ class Round : public UnaryPrimitive {
class Scan : public UnaryPrimitive {
public:
enum ReduceType { Max, Min, Sum, Prod };
enum ReduceType { Max, Min, Sum, Prod, LogAddExp };
explicit Scan(
Stream stream,
@@ -1763,6 +1763,9 @@ class Scan : public UnaryPrimitive {
case Max:
os << "Max";
break;
case LogAddExp:
os << "Logaddexp";
break;
}
}
bool is_equivalent(const Primitive& other) const override;