From ff312af52b87eaab9f55c83e7a179c79d15ca2e0 Mon Sep 17 00:00:00 2001 From: Astha Rai Date: Fri, 11 Jul 2025 01:28:39 +0000 Subject: [PATCH] fix default problem config parsing --- example/01_gemm/common.hpp | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/example/01_gemm/common.hpp b/example/01_gemm/common.hpp index 4104e83677..795e52b7a2 100644 --- a/example/01_gemm/common.hpp +++ b/example/01_gemm/common.hpp @@ -172,19 +172,17 @@ bool parse_cmd_args(int argc, if(argc >= 11) { - problem_size.Streamk_sel = std::stoi(argv[10]); - problem_size.Grid_size = std::stoi(argv[11]); + problem_size.NumSKBlocks = std::stoi(argv[10]); } } else { - std::cerr - << "arg1: verification (0=no, 1=CPU, 2=GPU, 3=CPU and GPU)" << std::endl - << "arg2: initialization (0=no init, 1=integer value, 2=decimal value)" << std::endl - << "arg3: time kernel (0=no, 1=yes)" << std::endl - << "arg4 to 9: M (256x), N(128x), K(32x), StrideA, StrideB, StrideC" << std::endl - << "arg10: stream-k select (-1: default config, 0: all DP, 1: 1-tile SK, 2: 2-tile SK)" - << "\narg11: Grid_size(-1 for max occupancy)" << std::endl; + std::cerr << "arg1: verification (0=no, 1=CPU, 2=GPU, 3=CPU and GPU)" << std::endl + << "arg2: initialization (0=no init, 1=integer value, 2=decimal value)" + << std::endl + << "arg3: time kernel (0=no, 1=yes)" << std::endl + << "arg4 to 9: M (256x), N(128x), K(32x), StrideA, StrideB, StrideC" << std::endl + << "arg10: NumSKBlocks" << std::endl; return false; }