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 1 Packages Projects Releases Wiki Activity
1,358 Commits 20 Branches 75 Tags
333ffea2730bb789a6eecdcaa78f1e808607c4be
Commit Graph

2 Commits

Author SHA1 Message Date
Cheng
f4c8888cbe [CUDA] Fix stride of singleton dims before passing to cuDNN (#2521) 2025-08-21 08:55:26 +09:00
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