7 Commits

Author	SHA1	Message	Date
Awni Hannun	dc371ae7a5	fix for max block dim (#2631)	2025-09-29 08:59:25 -07:00
Cheng	4822c3dbe9	[CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533) ... * Move DynamicSlice to gpu/primitives * Implement compute_dynamic_offset in CUDA	2025-08-26 07:31:39 +09:00
Angelos Katharopoulos	e397177f6e	Custom cuda kernel (#2517)	2025-08-20 17:20:22 -07:00
Cheng	828c5f1137	Use SmallVector for shapes and strides (#2454) ... * Use SmallVector for shapes and strides * Convert SmallVector to tuple	2025-08-05 09:41:03 +09:00
Awni Hannun	b9e88fb976	[CUDA] Fix segfault on exit (#2424) ... * fix cuda segfault on exit * comment	2025-07-27 08:08: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
Cheng	a4fc671d3e	CUDA backend: compile (#2276) ... * CUDA backend: compile * Rename kernels/ to device/	2025-06-12 17:08:39 -07:00