Skip to content

Commit d1c1b23

Browse files
LeungChiNanGong-airwey-code
authoredJan 24, 2024
ext test case (DeepLink-org#721)
* set flash-attention lib default path depends on python version. * rms_norm * rotary_emb * apply_penalty * destindex_copy_kv * token_attention * token_softmax_reducev * context_attention * start_loc start from not zero * mha add GLOBAL_STATE * multihead_attention_varlen --------- Co-authored-by: gqw <[email protected]> Co-authored-by: Gong-air <[email protected]> Co-authored-by: wey-code <[email protected]>
1 parent c0ffe2c commit d1c1b23

File tree

6 files changed

+555
-125
lines changed

6 files changed

+555
-125
lines changed
 

0 commit comments

Comments
 (0)
Please sign in to comment.