4 Commits

Author	SHA1	Message	Date
Cheng	2ca533b279	Fix compilation with CUDA 11 (#2331)	2025-07-07 20:00:43 -07:00
Awni Hannun	ec0d5db67b	[CUDA] Switch to CUDA graphs (#2317) ... * cuda graph prototype fix signal bug + start to add dependencies capture more capture more ops remaining ops fix reduce and rope deps add concurrent context try update, but not working cosistent topology order use node api use node api directly to reduce overhead fix bug use kernels in unary cache graph format fix synchronization format * comment	2025-07-02 15:59:13 -07:00
Angelos Katharopoulos	3d5e17e507	MLX_SWITCH macros to templates (#2320)	2025-07-01 01:33:44 -07:00
Angelos Katharopoulos	772f471ff2	[CUDA] Fix reductions (#2314)	2025-06-27 12:59:20 -07:00