refine mla v4 co#4083

feifei14119 wants to merge 4 commits into

ROCm:mainROCm/aiter:mainfrom

feifei14119:feiw/v4/mlafeifei14119/aiter:feiw/v4/mlaCopy head branch name to clipboard