github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-01 03:45:27 +00:00
Code Issues Packages Projects Releases Wiki Activity
feat/speculative-decoding
ColossalAI/extensions/softmax
History
Yuanheng ed5ebd1735 [Fix] resolve conflicts of merging main
2024-04-08 16:21:47 +08:00
..
__init__.py [devops] remove post commit ci (#5566) 2024-04-08 15:09:40 +08:00
scaled_masked_softmax_cuda.py refactor code 2024-03-11 17:06:57 +08:00
scaled_upper_triangle_masked_softmax_cuda.py refactor code 2024-03-11 17:06:57 +08:00