github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-19 10:32:56 +00:00
Code Issues Packages Projects Releases Wiki Activity
3,833 Commits 60 Branches 51 Tags
c4fe9e812ed6880b51bd3a7b6dc12cf81b402a21
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