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

fix

Browse Source
This commit is contained in:
Awni Hannun
2025-06-02 13:43:22 -07:00
parent f859e75f4f
commit d4aafb9161
2 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/ops.cpp
View File

@@ -2864,8 +2864,7 @@ array matmul(
}
// complex matmul using Karatsuba's Algorithm
if (a.dtype() == complex64 && b.dtype() == complex64) {
if (a.dtype() == complex64 || b.dtype() == complex64) {
// Extract real and imaginary parts
auto a_real = real(a, s);
auto a_imag = imag(a, s);