Skip to content

Commit 95b7551

Browse files
authored
Do not check event.is_signaled() in eval_impl (#2230)
1 parent db5a7c6 commit 95b7551

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

mlx/transforms.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -208,9 +208,7 @@ array eval_impl(std::vector<array> outputs, bool async) {
208208
// output arrays stream
209209
fences[it->second].wait(stream, in);
210210
} else if (in.event().valid()) {
211-
if (in.event().is_signaled()) {
212-
in.detach_event();
213-
} else if (in.event().stream() != stream) {
211+
if (in.event().stream() != stream) {
214212
// Use event to wait across async eval
215213
in.event().wait(stream);
216214
}

0 commit comments

Comments
 (0)