ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 13:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1973903f49947d33af88c41d173429a0057d8a59
composable_kernel/include/ck_tile/ops/fmha/kernel
History
Qianfeng 1973903f49 Hacking ck_tile fmha Dropout facility (#1344) 
* Add NullBlockDropout to be used when kHasDropout is false

* Change to BlockDropout::Run() for forward to reduce conditional checkings

* Re-format files

---------

Co-authored-by: PoYen, Chen <PoYen.Chen@amd.com>
2024-06-19 10:37:22 +08:00
..
fmha_bwd_kernel.hpp
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00
fmha_bwd_tile_partitioner.hpp
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00
fmha_fwd_kernel.hpp
Hacking ck_tile fmha Dropout facility (#1344)
2024-06-19 10:37:22 +08:00
fmha_fwd_tile_partitioner.hpp
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00