From f2510ae890b48653e3da9652839ccfb5fcea5c68 Mon Sep 17 00:00:00 2001 From: Alexander Weinrauch Date: Thu, 3 Oct 2024 14:33:56 +0000 Subject: [PATCH] Report default values in exported result to support changing them in the future --- .../tools/tune_gemm/test_regression.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/python/perf-kernels/tools/tune_gemm/test_regression.py b/python/perf-kernels/tools/tune_gemm/test_regression.py index ebac069809db..e309b9694bb3 100644 --- a/python/perf-kernels/tools/tune_gemm/test_regression.py +++ b/python/perf-kernels/tools/tune_gemm/test_regression.py @@ -104,14 +104,14 @@ def test_matmul_performance_regression(self, config, record_property): M, N, K, col_a, col_b, runConfig = tune_gemm.process_item(deepcopy(config)) - rotating_buffer_size = config.get('rotating_buffer_size', 0) - icache_flush = config.get('icache_flush', False) - iters = config.get('iters', 200) - init_type = config.get('init_type', 'randn') - - dtype_a = config.get('dtype_a', 'fp16') - dtype_b = config.get('dtype_b', 'fp16') - dtype_c = config.get('dtype_c', 'fp16') + rotating_buffer_size = config.setdefault('rotating_buffer_size', 0) + icache_flush = config.setdefault('icache_flush', False) + iters = config.setdefault('iters', 200) + init_type = config.setdefault('init_type', 'randn') + + dtype_a = config.setdefault('dtype_a', 'fp16') + dtype_b = config.setdefault('dtype_b', 'fp16') + dtype_c = config.setdefault('dtype_c', 'fp16') bias_vector = config.get('bias_vector', False) bias_size = M if bias_vector else 0