Fix building with CUDA < 12.8 (#2782)

2025-12-16 01:49:05 +08:00 · 2025-11-18 12:55:19 +09:00
parent 35f81728f1
commit 940f4c7818
40 changed files with 91 additions and 113 deletions
--- a/mlx/backend/cuda/custom_kernel.cpp
+++ b/mlx/backend/cuda/custom_kernel.cpp
@@ -289,7 +289,7 @@ void CustomKernel::eval_gpu(
      copies.emplace_back(init_value_.value(), out.dtype());
      fill_gpu(copies.back(), out, s);
    } else {
-      out.set_data(cu::malloc_async(out.nbytes(), encoder.stream()));
+      out.set_data(cu::malloc_async(out.nbytes(), encoder));
    }
  }