8 Commits

Author	SHA1	Message	Date
Angelos Katharopoulos	be9bc96da4	[CUDA] Matmul utils initial commit (#2441)	2025-08-01 14:22:25 -07:00
Cheng	254476718b	Remove the kernel arg from get_launch_args (#2437)	2025-07-30 11:43:02 +09:00
Awni Hannun	d107d8d495	add cuda gemv (#2400)	2025-07-22 08:24:13 -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
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
Awni Hannun	c552ff2451	[CUDA] Fix back-end bugs and enable corresponding tests (#2296) ... * Fix some cuda back-end bugs and enable corresponding tests * more fixes * enable more tests * format	2025-06-16 08:45:40 -07:00
Cheng	99c33d011d	rebase + nit (#2260) ... Co-authored-by: Awni Hannun <awni@apple.com>	2025-06-10 10:51:51 -07:00