Cheng 6f5874a2f2 [CUDA] Initial implementation of Convolution with cuDNN (#2385) ... * Link with cuDNN * Initial implementation * Remove backend apis * Fix recording cudnn conv * More unused backend apis * Fix C++ conv tests * include cudnn as python dep * Install libcudnn9-dev-cuda-12 in CI * cudnn only accepts contiguous inputs * Switch to backend apis * Plan needs to be kept alive * Turn off tf32 * Add cache * Test the native cuda graph api * Set cudnn stream before execution * Make LRUCache more like a normal container * Do error check for cublas handle * Zero-initilizing array * Use tf32 for conv * Skip TestConv.test_torch_conv_2D test --------- Co-authored-by: Awni Hannun <awni@apple.com>		2025-07-25 08:12:10 +09:00
..
repair_cuda.sh	[CUDA] Initial implementation of Convolution with cuDNN (#2385)	2025-07-25 08:12:10 +09:00
repair_linux.sh	Install linux with mlx[cuda] and mlx[cpu] (#2356)	2025-07-14 17:17:33 -07:00
repair_record.py	Install linux with mlx[cuda] and mlx[cpu] (#2356)	2025-07-14 17:17:33 -07:00