ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
44ec3b26d5c4edc17f8bc2cf6a8ef29df418b183
composable_kernel/include/ck_tile/ops
History
Qianfeng 44ec3b26d5 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>

[ROCm/composable_kernel commit: 1973903f49]
2024-06-19 10:37:22 +08:00
..
common
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
epilogue
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
fmha
Hacking ck_tile fmha Dropout facility (#1344)
2024-06-19 10:37:22 +08:00
gemm
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00
reduce/block
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
common.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
epilogue.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
fmha.hpp
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00
gemm.hpp
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00
reduce.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00