mirror of https://github.com/hpcaitech/ColossalAI
![]() * add rotary embedding kernel * add rotary_embedding_kernel * add fused rotary_emb and kvcache memcopy * add fused_rotary_emb_and_cache_kernel.cu * add fused_rotary_emb_and_memcopy * fix bugs in fused_rotary_emb_and_cache_kernel.cu * fix ci bugs * use vec memcopy and opt the gloabl memory access * fix code style * fix test_rotary_embdding_unpad.py * codes revised based on the review comments * fix bugs about include path * rm inline |
||
---|---|---|
.. | ||
inference.cpp | ||
layer_norm.cpp | ||
moe.cpp | ||
optimizer.cpp | ||
scaled_masked_softmax.cpp | ||
scaled_upper_triang_masked_softmax.cpp |