Optimize topksoftmax: top-K-only softmax + 32B vector loads#804

valarLip merged 1 commit intoROCm:mainROCm/aiter:mainfrom

CuiCu-618:cucui/opt_topksoftmaxCuiCu-618/aiter:cucui/opt_topksoftmaxCopy head branch name to clipboard