fix jit scan when output doesn't have primitive (#1190)

2025-12-16 01:49:05 +08:00 · 2024-06-06 07:24:58 -07:00
parent 496315fe1d
commit 578842954c
4 changed files with 15 additions and 7 deletions
--- a/mlx/backend/metal/nojit_kernels.cpp
+++ b/mlx/backend/metal/nojit_kernels.cpp
@@ -63,6 +63,7 @@ MTL::ComputePipelineState* get_scan_kernel(
    const std::string& kernel_name,
    bool,
    bool,
+    const std::string&,
    const array&,
    const array&) {
  return d.get_kernel(kernel_name);