shenggan 5c3843dc98 add colossalai kernel module (#55)		2021-12-21 12:19:52 +08:00
..
csrc	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
__init__.py	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
builder.py	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
layer_norm.py	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
multihead_attention.py	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
scaled_softmax.py	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00