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

Build CUDA release in Circle (#2306)

Browse Source 
* cuda release

* add license
This commit is contained in:
Awni Hannun
2025-06-19 15:26:36 -07:00
committed by GitHub
parent b3d7b85376
commit 76831ed83d
5 changed files with 144 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/cuda/device.cpp
View File

@@ -114,7 +114,7 @@ void CommandEncoder::synchronize() {
std::future<void> f = p->get_future();
add_completed_handler([p = std::move(p)]() { p->set_value(); });
worker_.end_batch();
worker_.commit();
commit();
f.wait();
}