github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-06 17:46:42 +00:00
Code Issues Packages Projects Releases Wiki Activity
3,495 Commits 59 Branches 51 Tags
229db4bc1623e74f2007074db35cb198e6c3b51a
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