Add Quantized Ops to the JIT (#1204)

* JIT for quantized ops * remove unused imports * address comments * fix imports * second attempt to fix imports --------- Co-authored-by: Alex Barron <abarron22@apple.com>
2025-12-16 01:49:05 +08:00 · 2024-06-12 09:47:12 -07:00
parent df964132fb
commit dd7d8e5e29
13 changed files with 1778 additions and 1948 deletions
--- a/mlx/backend/metal/CMakeLists.txt
+++ b/mlx/backend/metal/CMakeLists.txt
@@ -112,6 +112,7 @@ if (MLX_METAL_JIT)
    kernels/steel/defines.h
    kernels/steel/conv/loaders/loader_general.h
  )
+  make_jit_source(quantized)
 else()
  target_sources(
    mlx