fix malloc or wait deadlock (#1976)

2025-12-16 01:49:05 +08:00 · 2025-03-20 16:48:43 -07:00
parent 1177d28395
commit 7b7e2352cd
55 changed files with 201 additions and 217 deletions
--- a/mlx/backend/metal/scan.cpp
+++ b/mlx/backend/metal/scan.cpp
@@ -24,7 +24,7 @@ void Scan::eval_gpu(const std::vector<array>& inputs, array& out) {
      out.copy_shared_buffer(in);
    } else {
      out.set_data(
-          allocator::malloc_or_wait(in.data_size() * out.itemsize()),
+          allocator::malloc(in.data_size() * out.itemsize()),
          in.data_size(),
          in.strides(),
          in.flags());