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/compiled.cpp
+++ b/mlx/backend/cuda/compiled.cpp
@@ -298,7 +298,7 @@ void Compiled::eval_gpu(
  // Put outputs.
  compiled_allocate_outputs(
      inputs, outputs, is_constant_, contiguous, [&](auto n) {
-        return cu::malloc_async(n, encoder.stream());
+        return cu::malloc_async(n, encoder);
      });
  for (auto& x : outputs) {
    args.append(x);