zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-24 14:58:17 +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
1 changed files with 1 additions and 1 deletions
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(