ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI

History

ver217 12b4887097 [hotfix] fix CPUAdam kernel nullptr (#1410 )		2022-08-05 19:45:45 +08:00
..
csrc	[hotfix] fix CPUAdam kernel nullptr (#1410 )	2022-08-05 19:45:45 +08:00
__init__.py	…
layer_norm.py	[NFC] polish colossalai/kernel/cuda_native/layer_norm.py code style (#980 )	2022-05-17 10:25:06 +08:00
multihead_attention.py	…
scaled_softmax.py	…