apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 17:11:21 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity
Files
2359c8d21cce8d23dd0ec4e1d2a5d0ecdf86cc02
FastDeploy/custom_ops/gpu_ops/flash_mask_attn
T
History
yangjianfengo1 40f7f3e0d8 [New Feature] fa3 支持flash mask (#3184) 
* 支持flash mask

* 修改test_flash_mask

* 修改test.sh
2025-08-05 12:20:48 +08:00
..
flash_mask_attn_kernel.hpp
[New Feature] fa3 支持flash mask (#3184)
2025-08-05 12:20:48 +08:00
flash_mask_attn.cu
[New Feature] fa3 支持flash mask (#3184)
2025-08-05 12:20:48 +08:00
kernel_traits.h
[New Feature] fa3 支持flash mask (#3184)
2025-08-05 12:20:48 +08:00
mainloop_attn.hpp
[New Feature] fa3 支持flash mask (#3184)
2025-08-05 12:20:48 +08:00
softmax.hpp
[New Feature] fa3 支持flash mask (#3184)
2025-08-05 12:20:48 +08:00
utils.hpp
[New Feature] fa3 支持flash mask (#3184)
2025-08-05 12:20:48 +08:00