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

Perf regression fix (#2243)

Browse Source
This commit is contained in:
Angelos Katharopoulos 2025-06-03 17:55:12 -07:00 committed by GitHub
parent 85a8beb5e4
commit aede70e81d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/transforms.cpp
View File

@ -208,7 +208,9 @@ 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().stream() != stream) { if (in.event().is_signaled()) {
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);
} }

2
mlx/version.h
View File

@ -4,7 +4,7 @@
#define MLX_VERSION_MAJOR 0 #define MLX_VERSION_MAJOR 0
#define MLX_VERSION_MINOR 26 #define MLX_VERSION_MINOR 26
#define MLX_VERSION_PATCH 0 #define MLX_VERSION_PATCH 1
#define MLX_VERSION_NUMERIC \ #define MLX_VERSION_NUMERIC \
(100000 * MLX_VERSION_MAJOR + 1000 * MLX_VERSION_MINOR + MLX_VERSION_PATCH) (100000 * MLX_VERSION_MAJOR + 1000 * MLX_VERSION_MINOR + MLX_VERSION_PATCH)