zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-18 01:50:16 +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;
}
eval_impl(std::move(outputs), false).wait();
eval_impl(std::move(outputs), false).event().wait();
}
std::pair<std::vector<array>, std::vector<array>> vjp(