From 95b7551d65c24d65e0a3de2a656f11577499fd9f Mon Sep 17 00:00:00 2001 From: Cheng Date: Tue, 3 Jun 2025 05:23:34 +0900 Subject: [PATCH] Do not check event.is_signaled() in eval_impl (#2230) --- mlx/transforms.cpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/mlx/transforms.cpp b/mlx/transforms.cpp index 2d9942eda..3a02f39cb 100644 --- a/mlx/transforms.cpp +++ b/mlx/transforms.cpp @@ -208,9 +208,7 @@ array eval_impl(std::vector outputs, bool async) { // output arrays stream fences[it->second].wait(stream, in); } else if (in.event().valid()) { - if (in.event().is_signaled()) { - in.detach_event(); - } else if (in.event().stream() != stream) { + if (in.event().stream() != stream) { // Use event to wait across async eval in.event().wait(stream); }