lizhenyun01 aba4fc657f [Feature] support flash_mask_attention backend (#5134) ...

* [Feature] suppert flash_mask_attention backend

* fix unittest

* clean code

2025-11-28 10:12:16 +08:00

..

__init__.py

[Feature] support flash_mask_attention backend (#5134)

2025-11-28 10:12:16 +08:00

append_attention.py

…

flash_mask_attention.py

[Feature] support flash_mask_attention backend (#5134)

2025-11-28 10:12:16 +08:00

get_block_shape_and_split_kv_block.py

…

gqa_rope_write_cache.py

[Feature] support flash_mask_attention backend (#5134)

2025-11-28 10:12:16 +08:00

init_kv_signal_per_query.py

…

init_signal_layerwise.py

…

open_shm_and_get_meta_signal.py

…

pre_cache_len_concat.py

…