Add grid-stride + ROCm cap to permute_multi_embs_kernel (#6040) #9143

Job	Run time
build_artifact (x86, linux.4xlarge, 3.14, gcc)	3m 57s
build_artifact (arm, linux.arm64.m7g.4xlarge, 3.14, gcc)	3m 54s
benchmark_artifact (arm, linux.arm64.m7g.4xlarge, 30, 3.14, gcc)	1m 12s
benchmark_artifact (x86, linux.4xlarge, 20, 3.14, gcc)	1m 13s
	10m 16s

Provide feedback