Commit Graph

119 Commits (7486ed7d3a21ad35c4f465583426b25af6b33c04)

Author SHA1 Message Date
BoxiangW dfe423ae42 fix format (#572)
3 years ago
Xu Kai 2a915a8b62 fix format (#568)
3 years ago
YuliangLiu0306 9420d3ae31 fix format (#567)
3 years ago
Jie Zhu 0f1da44e5e [format]colossalai/kernel/cuda_native/csrc/layer_norm_cuda.cpp (#566)
3 years ago
coder-chin 5835631218 fix format (#564)
3 years ago
Luxios22 e014144c44 fix format (#565)
3 years ago
Ziyue Jiang 1762ba14ab fix format (#563)
3 years ago
HELSON b31daed4cf
fix bugs in CPU adam (#633)
3 years ago
KAIYUAN GAN 229382c844
[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/cuda_util.cu code stype (#625)
3 years ago
LuGY 6a3f9fda83
[cuda] modify the fused adam, support hybrid of fp16 and fp32 (#497)
3 years ago
ExtremeViscent eaac03ae1d [formart] format fixed for kernel\cuda_native codes (#335)
3 years ago
LuGY a3269de5c9 [zero] cpu adam kernel (#288)
3 years ago
1SAA 219df6e685 Optimized MoE layer and fixed some bugs;
3 years ago
アマデウス 9ee197d0e9 moved env variables to global variables; (#215)
3 years ago
HELSON 0f8c7f9804
Fixed docstring in colossalai (#171)
3 years ago
Frank Lee e2089c5c15
adapted for sequence parallel (#163)
3 years ago
Frank Lee f3802d6b06
fixed jit default setting (#154)
3 years ago
ver217 f68eddfb3d
refactor kernel (#142)
3 years ago
shenggan 5c3843dc98
add colossalai kernel module (#55)
3 years ago