Making large AI models cheaper, faster and more accessible
COMMITS
/ colossalai/kernel/__init__.py January 25, 2024
F
[feat] refactored extension module (#5298)
Frank Lee committed
January 8, 2024
X
[npu] use extension for op builder (#5172)
Xuanlei Zhao committed
September 14, 2023
Y
[hotfix] Fix import error: colossal.kernel without triton installed (#4722)
Yuanheng Zhao committed
September 11, 2023
January 6, 2023
F
[setup] support pre-build and jit-build of cuda kernels (#2374)
Frank Lee committed
January 3, 2023
J
[builder] MOE builder (#2277)
Jiarui Fang committed
December 30, 2022
J
[builder] builder for scaled_upper_triang_masked_softmax (#2234)
Jiarui Fang committed
December 27, 2022
J
[builder] multihead attn runtime building (#2203)
Jiarui Fang committed
December 23, 2022
J
[builder] unified cpu_optim fused_optim inferface (#2190)
Jiarui Fang committed
March 31, 2022
X
fix format (#568)
Xu Kai committed
January 13, 2022
V
refactor kernel (#142)
ver217 committed
December 21, 2021
S
add colossalai kernel module (#55)
shenggan committed