apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-04 16:22:57 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity
Files
c45ba7f39c53002b24b4f0d8813de9c06005e3ba
FastDeploy/custom_ops/gpu_ops/flash_mask_attn
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