[BUG][CK][MHA] Fix for MHA with softmax-sink - #4612

[BUG][CK][MHA] Fix for MHA with softmax-sink#4612

shurale-nkn wants to merge 3 commits into

mainROCm/aiter:mainfrom

knasyrov/mha_sinkROCm/aiter:knasyrov/mha_sinkCopy head branch name to clipboard