From 5548fcc96d4abcebb7477dd71b332dd539e1af70 Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Wed, 18 Dec 2024 12:25:16 -0800
Subject: [PATCH] fix synch race (#1719)

---
 mlx/transforms.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlx/transforms.cpp b/mlx/transforms.cpp
index e1b94aa28..88bfebc1b 100644
--- a/mlx/transforms.cpp
+++ b/mlx/transforms.cpp
@@ -268,7 +268,7 @@ void eval(std::vector<array> outputs) {
     return;
   }
 
-  eval_impl(std::move(outputs), false).wait();
+  eval_impl(std::move(outputs), false).event().wait();
 }
 
 std::pair<std::vector<array>, std::vector<array>> vjp(