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/cpu/softmax.cpp
+++ b/mlx/backend/cpu/softmax.cpp
@@ -129,7 +129,7 @@ void Softmax::eval_cpu(const std::vector<array>& inputs, array& out) {
        out.copy_shared_buffer(x);
      } else {
        out.set_data(
-            allocator::malloc_or_wait(x.data_size() * x.itemsize()),
+            allocator::malloc(x.data_size() * x.itemsize()),
            x.data_size(),
            x.strides(),
            x.flags());