mlx/mlx/backend/accelerate
Kashif Rasul 3ddc07e936
Eigenvalues and eigenvectors (#1334)
* initial eigvalsh

* add compute_vectors

* add compute_vectors_

* return a pair

* add eigh to return only eigenvectors

* fixed typo

* merge merge Eighvalsh and Eigh into a single primitive

* use the same primate with the flag

* fix primatives

* use MULTI

* fix eval_gpu

* fix decleration

* rename EighPrimitive to Eigh

* tests

* tests

* fix rebase and format

* cleanup lapack

* format

* add cblas.h

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-10-22 12:18:48 -07:00
..
CMakeLists.txt Chore: add pre-commit hook for cmake (#1362) 2024-09-16 12:53:01 -07:00
conv.cpp Accelerate import updates for iOS (#1227) 2024-06-26 09:01:50 -07:00
matmul.cpp Accelerate import updates for iOS (#1227) 2024-06-26 09:01:50 -07:00
primitives.cpp Eigenvalues and eigenvectors (#1334) 2024-10-22 12:18:48 -07:00
quantized.cpp Batched Quantized Matmul + Fast Small QMV (#1503) 2024-10-21 16:23:17 -07:00
reduce.cpp Accelerate import updates for iOS (#1227) 2024-06-26 09:01:50 -07:00
softmax.cpp consistently handle all -inf in softmax (#1470) 2024-10-08 09:54:02 -07:00
utils.h Accelerate import updates for iOS (#1227) 2024-06-26 09:01:50 -07:00