23 Commits

Author	SHA1	Message	Date
Angelos Katharopoulos	4a9b29a875	MoE backward improvements (#2335)	2025-07-07 17:59:53 -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	b8022c578a	divmod, partition, sort fixes (#2302)	2025-06-16 18:49:32 -07:00
Angelos Katharopoulos	580776559b	RoPE for CUDA (#2293) ... * First working CUDA rope * Fix random	2025-06-15 06:08:07 -07:00
Cheng	c8b4787e4e	CUDA backend: indexing ops (#2277)	2025-06-12 21:44:19 -07:00
Awni Hannun	2188199ff8	[CUDA] ternary with select op (#2283) ... * cuda ternary with select op * comment + fix * fix	2025-06-12 20:24:43 -07:00
Awni Hannun	918761a25a	[CUDA] RMSNorm and VJP (#2280) ... * rms norm start * nit	2025-06-12 17:09:49 -07:00
Cheng	a4fc671d3e	CUDA backend: compile (#2276) ... * CUDA backend: compile * Rename kernels/ to device/	2025-06-12 17:08:39 -07:00
Awni Hannun	f5f65ef48c	Make sliceUpdate general (#2282) ... * Make sliceUpdate general * fix	2025-06-12 16:48:54 -07:00
Cheng	d7e680ffe4	CUDA backend: layernorm (#2271)	2025-06-11 15:48:32 -07:00
Cheng	c371baf53a	CUDA backend: softmax (#2272)	2025-06-11 13:55:22 -07:00
Cheng	ccf78f566c	CUDA backend: argreduce (#2270)	2025-06-11 13:26:17 -07:00
Cheng	c9fa68664a	CUDA backend: reduce (#2269)	2025-06-11 11:22:25 -07:00
Cheng	7c4eb5d03e	CUDA backend: random (#2261)	2025-06-10 08:59:56 -07:00
Cheng	bae9a6b404	CUDA backend: sort (#2262) ... Co-authored-by: Awni Hannun <awni@apple.com>	2025-06-10 08:59:47 -07:00
Cheng	7ebb2e0193	CUDA backend: binary ops (#2259)	2025-06-10 06:37:40 -07:00
Cheng	f8bad60609	CUDA backend: unary ops (#2158)	2025-06-09 06:45:08 -07:00
Cheng	24f89173d1	CUDA backend: matmul (#2241)	2025-06-06 12:24:04 -07:00
Cheng	1b021f6984	Fast primitives decide when to use the fallback (#2216)	2025-06-02 13:26:37 -07:00
Cheng	f76ee1ffd2	Move some dims utils to common (#2223)	2025-05-29 06:48:30 -07:00
Cheng	7d4b378952	Include cuda_bf16.h for bfloat16 overloads (#2192) ... * Include cuda_bf16.h for bfloat16 overloads * Add NO_GPU_MULTI(Eig) in cuda backend	2025-05-16 06:44:42 -07:00
Cheng	0cae0bdac8	CUDA backend: backbone (#2075)	2025-05-06 21:26:46 -07:00