apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity
Files
8e0f4dfd0cf70a0a67aca59629089b4a193404bb
FastDeploy/custom_ops/gpu_ops/flash_mask_attn
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