mlx/mlx/backend
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
..
common [CUDA] Switch to CUDA graphs (#2317) 2025-07-02 15:59:13 -07:00
cpu CUDA backend: unary ops (#2158) 2025-06-09 06:45:08 -07:00
cuda [CUDA] Switch to CUDA graphs (#2317) 2025-07-02 15:59:13 -07:00
gpu Make sliceUpdate general (#2282) 2025-06-12 16:48:54 -07:00
metal Fix complex power and print (#2286) 2025-06-13 11:13:00 -07:00
no_cpu non-symmetric eig and eigh (#2188) 2025-05-15 13:01:44 -07:00
no_gpu start cuda circle config (#2256) 2025-06-10 21:19:47 -07:00