mlx/mlx/backend/gpu
Cheng 1ba18ff7d9
[CUDA] Fix conv grads with groups (#2495)
* Put reshape utils in one file

* [CUDA] Fix conv grads with groups

* Put the reshape utils in gpu/copy.h
2025-08-16 10:09:18 +09:00
..
available.h Generalize gpu backend (#2138) 2025-04-30 09:08:17 -07:00
CMakeLists.txt Move common gpu primitives to backend/gpu (#2145) 2025-05-05 13:45:29 -07:00
copy.cpp [CUDA] Fix conv grads with groups (#2495) 2025-08-16 10:09:18 +09:00
copy.h [CUDA] Fix conv grads with groups (#2495) 2025-08-16 10:09:18 +09:00
eval.h Generalize gpu backend (#2138) 2025-04-30 09:08:17 -07:00
primitives.cpp [CUDA] Fix conv grads with groups (#2495) 2025-08-16 10:09:18 +09:00
slicing.cpp Move common gpu primitives to backend/gpu (#2145) 2025-05-05 13:45:29 -07:00
slicing.h Move common gpu primitives to backend/gpu (#2145) 2025-05-05 13:45:29 -07:00