apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity
Files
c694fa2879f118a793c830cf19f58cb3ba7ea788
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