github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 10:54:49 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
d83c633ca63c4eef49f3473aa998515fa5ca573f
ColossalAI/extensions/csrc/cuda/scaled_masked_softmax.cpp
Frank Lee 7cfed5f076 [feat] refactored extension module (#5298) 
* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish
2024-01-25 17:01:48 +08:00

2.6 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink