zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 18:11:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix synch race (#1719)

Browse Source
This commit is contained in:
Awni Hannun 2024-12-18 12:25:16 -08:00 committed by GitHub
parent 070bd433ab
commit 5548fcc96d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/transforms.cpp
View File

@ -268,7 +268,7 @@ void eval(std::vector<array> outputs) {
return; return;
} }
eval_impl(std::move(outputs), false).wait(); eval_impl(std::move(outputs), false).event().wait();
} }
std::pair<std::vector<array>, std::vector<array>> vjp( std::pair<std::vector<array>, std::vector<array>> vjp(