From 7a59980e9ffb523169f174ea253b2a3acbbe0b00 Mon Sep 17 00:00:00 2001 From: Jimniu Date: Wed, 13 Aug 2025 19:06:08 -0400 Subject: [PATCH] Jimniu/tile_example_flatmm_basic fix (#2680) * Add stride_b validation * run clang-format [ROCm/composable_kernel commit: 753b6227c507c45f4838971067f3cf158a392d55] --- example/ck_tile/18_flatmm/flatmm_basic.cpp | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/example/ck_tile/18_flatmm/flatmm_basic.cpp b/example/ck_tile/18_flatmm/flatmm_basic.cpp index 475a0c7bf3..50bf791207 100644 --- a/example/ck_tile/18_flatmm/flatmm_basic.cpp +++ b/example/ck_tile/18_flatmm/flatmm_basic.cpp @@ -217,6 +217,17 @@ int run_flatmm_example(int argc, char* argv[]) std::string data_type = arg_parser.get_str("prec"); std::string a_layout = arg_parser.get_str("a_layout"); std::string b_layout = arg_parser.get_str("b_layout"); + + int k = arg_parser.get_int("k"); + int stride_b = arg_parser.get_int("stride_b"); + + if(b_layout == "C" && stride_b > k) + { + throw std::runtime_error( + "For ColumnMajor layout, StrideB must be smaller than or equal to K (" + + std::to_string(k) + ")"); + } + if(a_layout == "R" && b_layout == "C") {