zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 09:21:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Do not check event.is_signaled() in eval_impl (#2230)

Browse Source
This commit is contained in:
Cheng 2025-06-03 05:23:34 +09:00 committed by GitHub
parent db5a7c6192
commit 95b7551d65
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/transforms.cpp
View File

@ -208,9 +208,7 @@ array eval_impl(std::vector<array> outputs, bool async) {
// output arrays stream // output arrays stream
fences[it->second].wait(stream, in); fences[it->second].wait(stream, in);
} else if (in.event().valid()) { } else if (in.event().valid()) {
if (in.event().is_signaled()) { if (in.event().stream() != stream) {
in.detach_event();
} else if (in.event().stream() != stream) {
// Use event to wait across async eval // Use event to wait across async eval
in.event().wait(stream); in.event().wait(stream);
} }