From 9c2c2e7b636af5d6ba51fecff3f628d1cd728d57 Mon Sep 17 00:00:00 2001 From: Illia Silin <98187287+illsilin@users.noreply.github.com> Date: Fri, 31 Jan 2025 09:42:43 -0800 Subject: [PATCH] fix ck_tile gemm scripts (#1851) [ROCm/composable_kernel commit: 2ab8bf4c12ba99854afc406ad24626080ee1acd1] --- example/ck_tile/03_gemm/script/benchmark_basic.sh | 4 ++-- example/ck_tile/03_gemm/script/benchmark_mem_pipeline.sh | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/example/ck_tile/03_gemm/script/benchmark_basic.sh b/example/ck_tile/03_gemm/script/benchmark_basic.sh index f5473e46f4..6c6049ef8b 100755 --- a/example/ck_tile/03_gemm/script/benchmark_basic.sh +++ b/example/ck_tile/03_gemm/script/benchmark_basic.sh @@ -1,12 +1,12 @@ #!/bin/sh EXE="$(find . -name tile_example_gemm_basic -type f | head -n 1)" -VALID=0 +VALID=1 for b_matrix_layout in "R" "C"; do for m in "64" "512" "1024" "2048"; do for n in "512" "1024" "2048"; do for k in "64" "512" "1024" "2048"; do - $EXE -prec=fp16 -b=1 -m=$m -n=$n -k=$k -a_layout="R" -b_layout="$b_matrix_layout" -c_layout="R" -v=$VALID + $EXE -prec=fp16 -m=$m -n=$n -k=$k -a_layout="R" -b_layout="$b_matrix_layout" -c_layout="R" -v=$VALID done done done diff --git a/example/ck_tile/03_gemm/script/benchmark_mem_pipeline.sh b/example/ck_tile/03_gemm/script/benchmark_mem_pipeline.sh index a3029cbeb5..8ff7d7ad44 100755 --- a/example/ck_tile/03_gemm/script/benchmark_mem_pipeline.sh +++ b/example/ck_tile/03_gemm/script/benchmark_mem_pipeline.sh @@ -1,12 +1,12 @@ #!/bin/sh EXE="$(find . -name tile_example_gemm_universal -type f | head -n 1)" -VALID=0 +VALID=1 for b_matrix_layout in "R" "C"; do for m in "64" "512" "1024" "2048"; do for n in "512" "1024" "2048"; do for k in "64" "512" "1024" "2048"; do - $EXE -prec=fp16 -b=1 -m=$m -n=$n -k=$k -a_layout="R" -b_layout="$b_matrix_layout" -c_layout="R" -v=$VALID + $EXE -prec=fp16 -m=$m -n=$n -k=$k -a_layout="R" -b_layout="$b_matrix_layout" -c_layout="R" -v=$VALID done done done