mirror of https://github.com/hpcaitech/ColossalAI
![]() * update flash-context-attention * adding kernels * fix * reset * add build script * add building process * add llama2 exmaple * add colossal-llama2 test * clean * fall back test setting * fix test file * clean * clean * clean --------- Co-authored-by: cuiqing.li <lixx336@gmail.com> |
||
---|---|---|
.. | ||
kernel_utils.py | ||
test_bloom_context_attention.py | ||
test_copy_kv_dest.py | ||
test_layernorm_triton.py | ||
test_llama_act_combine.py | ||
test_llama_context_attention.py | ||
test_self_attention_nonfusion.py | ||
test_softmax.py | ||
test_token_attn_fwd.py | ||
test_token_softmax.py |