Implement the 'where' primitive for conditional selection (#664)

2025-12-16 01:49:05 +08:00 · 2024-02-22 18:10:48 -05:00
parent ad4a45e615
commit 126c9869c8
23 changed files with 991 additions and 56 deletions
--- a/mlx/backend/metal/kernels/CMakeLists.txt
+++ b/mlx/backend/metal/kernels/CMakeLists.txt
@@ -27,6 +27,7 @@ set(
  "scan"
  "softmax"
  "sort"
+  "ternary"
  "unary"
  "gather"
  "scatter"