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 outputs) { return; } - eval_impl(std::move(outputs), false).wait(); + eval_impl(std::move(outputs), false).event().wait(); } std::pair, std::vector> vjp(