leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
291300ffffa3533a78ee104f08a8490a29ce9ccb
cutlass/examples/python/CuTeDSL/hopper/cta_norm.py
Yihan Chen 291300ffff [CuTeDSL] implment a cta-level norm example (both layernorm and rmsnorm) (#3009) 
* kernel impl

* add copyright
2026-02-14 17:54:03 +08:00

16 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink