apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-25 09:57:51 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity
Files
9f4977eb7411a7e195cc5fca2aef4a9727446785
FastDeploy/custom_ops/gpu_ops/flash_mask_attn
T
History
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
..
flash_mask_attn_kernel.hpp
[Feature] support flash_mask_attention backend (#5134)
2025-11-28 10:12:16 +08:00
flash_mask_attn.cu
[Feature] support flash_mask_attention backend (#5134)
2025-11-28 10:12:16 +08:00
kernel_traits.h
[Feature] support flash_mask_attention backend (#5134)
2025-11-28 10:12:16 +08:00
mainloop_attn.hpp
format flash_mask_attn
2025-11-18 17:18:12 +08:00
softmax.hpp
format flash_mask_attn
2025-11-18 17:18:12 +08:00
utils.hpp
format flash_mask_attn
2025-11-18 17:18:12 +08:00