ming1753 734fbcffde [BugFix] Fix Async D2H copy bug & flash mash atten cache V out of bound bug (#7221)

2026-04-10 11:31:51 +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

[BugFix] Fix Async D2H copy bug & flash mash atten cache V out of bound bug (#7221)

2026-04-10 11:31:51 +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