ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-07 00:04:37 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
070e89d7adffb6dbd791f973dca249489330ea97
composable_kernel/include/ck_tile/ops/fmha
History
Dan Yao 1da802bdf2 Fix FA bwd alibi+causal NaN errors (#1352) 
* fix bwd alibi nan error

* fix datatype

---------

Co-authored-by: danyao12 <danyao12>
2024-06-20 09:50:53 -05:00
..
block
Fix FA bwd alibi+causal NaN errors (#1352)
2024-06-20 09:50:53 -05:00
kernel
Fix in dropout lambda to avoid the compiling issue on some docker/compiler envs (#1350)
2024-06-20 11:36:42 +08:00
pipeline
Fix FA bwd alibi+causal NaN errors (#1352)
2024-06-20 09:50:53 -05:00