Add matrix inversion primitive (#822)

2025-10-22 19:28:14 +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/metal/primitives.cpp
+++ b/mlx/backend/metal/primitives.cpp
@@ -900,4 +900,8 @@ void SVD::eval_gpu(
  throw std::runtime_error("[SVD::eval_gpu] Metal SVD NYI.");
 }

+void Inverse::eval_gpu(const std::vector<array>& inputs, array& output) {
+  throw std::runtime_error("[Inverse::eval_gpu] Metal inversion NYI.");
+}
+
 } // namespace mlx::core