ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 13:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
cb13839425e0ec4dfff5b8138104ee3e3183d050
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