xiaoxiaohehe001 51efe27d76 [BugFix] Fix batch_size derivation and relax shape checks in SM90 flash_mask_attn (#7210) ...

* [BugFix] fix_flash_mask_attn_sm90

* [BugFix] fix_flash_mask_attn_sm90

* [BugFix] Fix batch_size derivation and relax shape checks in SM90 flash_mask_attn

* [BugFix] Fix batch_size derivation and relax shape checks in SM90 flash_mask_attn

2026-04-09 11:05:10 +08:00

..

flash_mask_attn_kernel.hpp

[Feature] support flash_mask_attention backend (#5134)

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

flash_mask_attn.cu

[BugFix] Fix batch_size derivation and relax shape checks in SM90 flash_mask_attn (#7210)

2026-04-09 11:05:10 +08:00

kernel_traits.h

[Feature] support flash_mask_attention backend (#5134)

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

mainloop_attn.hpp

[Bug Fix] fix mask attention (#6216)

2026-01-26 07:46:26 -08:00

softmax.hpp

[BugFix] Fix kv cache int8 dynamic quant on flash and flash_mask backend (#7028)

2026-03-30 11:17:15 +08:00

utils.hpp

remove unneeded para from flash_mask_attention (#6218)

2026-01-27 14:04:27 +08:00