zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
1,329 Commits 20 Branches 75 Tags
a22d0bf273716144b41a2f3a42a737e514b62ed4
Commit Graph

3 Commits

Author SHA1 Message Date
Awni Hannun
1e496ddb82 [CUDA] Simplify allocator (#2392) 
* simplify allocator and fixe race with small pool

* Don't use shared event in worker

* use cuda buffer in small pool

* comment

* comment
 2025-07-22 08:24:01 -07:00
Awni Hannun
cad5c0241c [CUDA] synch properly waits for all tasks to finish and clear (#2303) 
* cuda synch properly waits for all tasks to finish and clear

* fix copy
 2025-06-17 12:03:25 -07:00
Cheng
0cae0bdac8 CUDA backend: backbone (#2075) 2025-05-06 21:26:46 -07:00