github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-20 04:32:47 +00:00
Code Issues Packages Projects Releases Wiki Activity
dev/zero-offload
ColossalAI/extensions/softmax
History
pre-commit-ci[bot] df612434c9 [pre-commit.ci] auto fixes from pre-commit.com hooks 
for more information, see https://pre-commit.ci
2024-06-14 16:27:46 +08:00
..
__init__.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2024-06-14 16:27:46 +08:00
scaled_masked_softmax_cuda.py [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
scaled_upper_triangle_masked_softmax_cuda.py [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00