github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-29 21:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
3,860 Commits 60 Branches 51 Tags
4fc92aa77dafd4a8253ed5ea4c16f090b44d2744
Commit Graph

2 Commits

Author SHA1 Message Date
Steve Luo
ccf72797e3 feat baichuan2 rmsnorm whose hidden size equals to 5120 (#5611) 2024-04-19 15:34:53 +08:00
Steve Luo
f7aecc0c6b feat rmsnorm cuda kernel and add unittest, benchmark script (#5417) 2024-03-08 16:21:12 +08:00