Commit d1c1b23
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- diopi_test/python
- configs
- conformance
- impl/ascend
- scripts/ci
6 files changed
+555
-125
lines changed
0 commit comments