Improve ROCm SDPA fallback behavior and MI350 support#1

keithloweryamd wants to merge 2 commits intoandyluo7:masterandyluo7/autoresearch:masterfrom

keithloweryamd:rocm-upstream-candidatekeithloweryamd/autoresearch:rocm-upstream-candidateCopy head branch name to clipboard