Add perf CI configs and fallbacks for gemm. Remove old regression tes…

…ting script
ROCm · Nov 13, 2024 · ab955c9 · ab955c9
1 parent 2a715fe
commit ab955c9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 172 deletions.
diff --git a/...nels/tools/tune_gemm/perf_ci_configs.yaml → ...regression_ci/fallback_tuned_configs.yaml b/...nels/tools/tune_gemm/perf_ci_configs.yaml → ...regression_ci/fallback_tuned_configs.yaml
@@ -1,4 +1,5 @@
-# M // BLOCK_M * N // BLOCK_N % 304 == 0
+# Fallback tuned configs for CI in case the weekly tuning has not been run yet
+
 ## 1 workgroup / CU
 - {'M': 4864, 'N': 4096, 'K': 4096, 'rowMajorA': 'T', 'rowMajorB': 'N', 'BLOCK_SIZE_M': 256, 'BLOCK_SIZE_N': 256, 'BLOCK_SIZE_K': 64, 'GROUP_SIZE_M': 4, 'SPLIT_K': 1, 'num_warps': 8, 'num_stages': 2, 'waves_per_eu': 0, 'matrix_instr_nonkdim': 16, 'kpack': 2}
 - {'M': 4864, 'N': 4096, 'K': 4160, 'rowMajorA': 'T', 'rowMajorB': 'N', 'BLOCK_SIZE_M': 256, 'BLOCK_SIZE_N': 256, 'BLOCK_SIZE_K': 64, 'GROUP_SIZE_M': 4, 'SPLIT_K': 1, 'num_warps': 8, 'num_stages': 2, 'waves_per_eu': 0, 'matrix_instr_nonkdim': 16, 'kpack': 2}

diff --git a/...ls/tools/tune_gemm/ci_tuning_configs.yaml → ...perf_regression_ci/tuning_gemm_sizes.yaml b/...ls/tools/tune_gemm/ci_tuning_configs.yaml → ...perf_regression_ci/tuning_gemm_sizes.yaml
@@ -1,3 +1,6 @@
+# If new gemm sizes/configs are added please provide a fallback tuned config in fallback_configs.yaml.
+# The fallback configs will be used in case the weekly tuning has not been run
+#
 # M // BLOCK_M * N // BLOCK_N % 304 == 0
 ## 1 workgroup / CU
 - {'M': 4864, 'N': 4096, 'K': 4096, 'rowMajorA': 'T', 'rowMajorB': 'N'}

diff --git a/python/perf-kernels/tools/tune_gemm/test_regression.py b/python/perf-kernels/tools/tune_gemm/test_regression.py