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

Add conjugate operator (#1100)

Browse Source 
* cpu and gpu impl

* add mx.conj and array.conj()

---------

Co-authored-by: Alex Barron <abarron22@apple.com>
This commit is contained in:
Alex Barron
2024-05-10 07:22:20 -07:00
committed by GitHub
parent 8bd6bfa4b5
commit 2e158cf6d0
17 changed files with 143 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
mlx/compile.cpp
View File

@@ -23,16 +23,17 @@ bool is_unary(const Primitive& p) {
typeid(p) == typeid(ArcSinh) || typeid(p) == typeid(ArcTan) ||
typeid(p) == typeid(ArcTanh) || typeid(p) == typeid(AsType) ||
typeid(p) == typeid(Ceil) || typeid(p) == typeid(Cos) ||
typeid(p) == typeid(Cosh) || typeid(p) == typeid(Remainder) ||
typeid(p) == typeid(Erf) || typeid(p) == typeid(ErfInv) ||
typeid(p) == typeid(Exp) || typeid(p) == typeid(Floor) ||
typeid(p) == typeid(Log) || typeid(p) == typeid(Log1p) ||
typeid(p) == typeid(LogicalNot) || typeid(p) == typeid(Negative) ||
typeid(p) == typeid(Round) || typeid(p) == typeid(Sigmoid) ||
typeid(p) == typeid(Sign) || typeid(p) == typeid(Sin) ||
typeid(p) == typeid(Sinh) || typeid(p) == typeid(Square) ||
typeid(p) == typeid(Sqrt) || typeid(p) == typeid(Tan) ||
typeid(p) == typeid(Tanh) || typeid(p) == typeid(Expm1));
typeid(p) == typeid(Conjugate) || typeid(p) == typeid(Cosh) ||
typeid(p) == typeid(Remainder) || typeid(p) == typeid(Erf) ||
typeid(p) == typeid(ErfInv) || typeid(p) == typeid(Exp) ||
typeid(p) == typeid(Floor) || typeid(p) == typeid(Log) ||
typeid(p) == typeid(Log1p) || typeid(p) == typeid(LogicalNot) ||
typeid(p) == typeid(Negative) || typeid(p) == typeid(Round) ||
typeid(p) == typeid(Sigmoid) || typeid(p) == typeid(Sign) ||
typeid(p) == typeid(Sin) || typeid(p) == typeid(Sinh) ||
typeid(p) == typeid(Square) || typeid(p) == typeid(Sqrt) ||
typeid(p) == typeid(Tan) || typeid(p) == typeid(Tanh) ||
typeid(p) == typeid(Expm1));
}
bool is_binary(const Primitive& p) {