ColossalAI

Author	SHA1	Message	Date
Frank Lee	40d376c566	[setup] support pre-build and jit-build of cuda kernels (#2374 ) * [setup] support pre-build and jit-build of cuda kernels * polish code * polish code * polish code * polish code * polish code * polish code	2023-01-06 20:50:26 +08:00
Jiarui Fang	16cc8e6aa7	[builder] MOE builder (#2277 )	2023-01-03 20:29:39 +08:00
Jiarui Fang	db4cbdc7fb	[builder] builder for scaled_upper_triang_masked_softmax (#2234 )	2022-12-30 09:58:00 +08:00
Jiarui Fang	1cb532ffec	[builder] multihead attn runtime building (#2203 ) * [hotfix] correcnt cpu_optim runtime compilation * [builder] multihead attn * fix bug * fix a bug	2022-12-27 16:06:09 +08:00
Jiarui Fang	355ffb386e	[builder] unified cpu_optim fused_optim inferface (#2190 )	2022-12-23 20:57:41 +08:00
Xu Kai	2a915a8b62	fix format (#568 )	2022-04-06 11:40:59 +08:00
ver217	f68eddfb3d	refactor kernel (#142 )	2022-01-13 16:47:17 +08:00
shenggan	5c3843dc98	add colossalai kernel module (#55 )	2021-12-21 12:19:52 +08:00