github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/kernel
History
xcnick 85178a397a
[hotfix] fix error for torch 2.0 (#2243) 		2022-12-30 23:11:55 +08:00
..
cuda_native [hotfix] fix error for torch 2.0 (#2243) 2022-12-30 23:11:55 +08:00
jit [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
op_builder [builder] builder for scaled_upper_triang_masked_softmax (#2234) 2022-12-30 09:58:00 +08:00
__init__.py [builder] builder for scaled_upper_triang_masked_softmax (#2234) 2022-12-30 09:58:00 +08:00