ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
c30ad40dfba2067bf910b849825f63d44dadc5a5
composable_kernel/include/ck_tile/ops
History
Dan Yao c30ad40dfb Fix FA bwd alibi+causal NaN errors (#1352) 
* fix bwd alibi nan error

* fix datatype

---------

Co-authored-by: danyao12 <danyao12>

[ROCm/composable_kernel commit: 1da802bdf2]
2024-06-20 09:50:53 -05: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
Fix FA bwd alibi+causal NaN errors (#1352)
2024-06-20 09:50:53 -05: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