Fix bug in momentum type declaration in HIP TBE kernel#147

aryaman-gupta wants to merge 5 commits intoaryaman/upstreamROCm/FBGEMM:aryaman/upstreamfrom

aryaman/hip-tbe-momentum-fixROCm/FBGEMM:aryaman/hip-tbe-momentum-fixCopy head branch name to clipboard