[ROCM] fix layout for WMMA_F16_16x16x16_F16 intrinsic (#18206) #2775

Job	Run time
setup / setup	14s
build_for_benchmarks / build_all	7m 46s
build_benchmark_tools / build_benchmark_tools (linux, x86_64, gcr.io/iree-oss/base-bleeding-edge@sha256:cf2e78194e64fd016...	2m 1s
build_benchmark_tools / build_benchmark_tools (linux, riscv_64, gcr.io/iree-oss/riscv@sha256:62e87bad3405d691ddba6f9be0ef...	3m 3s
build_benchmark_tools / build_benchmark_tools (android, armv8.2-a, gcr.io/iree-oss/android@sha256:66b92a1c920588a73d03316...	2m 15s
build_e2e_test_artifacts / build_e2e_test_artifacts	11m 5s
test_benchmark_suites (linux, riscv_64, gcr.io/iree-oss/riscv@sha256:62e87bad3405d691ddba6f9be0ef...	3m 15s
test_benchmark_suites (linux, riscv_32, gcr.io/iree-oss/riscv@sha256:62e87bad3405d691ddba6f9be0ef...	2m 59s
test_benchmark_suites (linux, x86_64, gcr.io/iree-oss/base@sha256:dc314b4fe30fc1315742512891357bf...	2m 36s
execution_benchmarks / generate_matrix	16s
compilation_benchmarks / compilation_benchmarks	36s
execution_benchmarks / run_benchmarks (c2-standard-60, linux, x86_64, 0, 2)	10m 3s
execution_benchmarks / run_benchmarks (c2-standard-60, linux, x86_64, 1, 2)	10m 14s
process_benchmark_results	5m 34s
benchmark_summary	5s
	1h 2m 2s

Provide feedback