zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
f7c11b965e8a467abaa4b2f0bb1e62fd182fc60f
mlx/mlx/backend/cuda/device.cpp
Cheng 65d0d40232 Split cuDNN helpers into a separate header (#2491) 
* Add RAII managed CudaGraph class

* Implement forward rms_norm with cuDNN

* Revert back to old rms norm kernel
2025-08-20 09:29:28 +09:00

9.7 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink