Add matrix inversion primitive (#822)

2025-12-16 01:49:05 +08:00 · 2024-03-15 14:34:36 +01:00
parent 19ec023256
commit eaba55c9bf
13 changed files with 204 additions and 4 deletions
--- a/mlx/backend/common/CMakeLists.txt
+++ b/mlx/backend/common/CMakeLists.txt
@@ -54,6 +54,7 @@ target_sources(
  ${CMAKE_CURRENT_SOURCE_DIR}/load.cpp
  ${CMAKE_CURRENT_SOURCE_DIR}/qrf.cpp
  ${CMAKE_CURRENT_SOURCE_DIR}/svd.cpp
+  ${CMAKE_CURRENT_SOURCE_DIR}/inverse.cpp
  ${CMAKE_CURRENT_BINARY_DIR}/compiled_preamble.cpp
 )