From 4978c785dad4e6bd199fb465e70e1d01e3aaaa37 Mon Sep 17 00:00:00 2001 From: coderfeli Date: Thu, 7 Aug 2025 04:39:37 +0000 Subject: [PATCH] reduce instance --- example/ck_tile/18_flatmm/flatmm_basic.cpp | 49 +++++++--------------- 1 file changed, 16 insertions(+), 33 deletions(-) diff --git a/example/ck_tile/18_flatmm/flatmm_basic.cpp b/example/ck_tile/18_flatmm/flatmm_basic.cpp index 60dced003b..39bd556a28 100644 --- a/example/ck_tile/18_flatmm/flatmm_basic.cpp +++ b/example/ck_tile/18_flatmm/flatmm_basic.cpp @@ -393,7 +393,7 @@ auto create_args(int argc, char* argv[]) .insert("timer", "gpu", "gpu:gpu timer, cpu:cpu timer") .insert("split_k", "1", "splitK value") .insert("init", "0", "0:random, 1:linear, 2:constant(1)") - .insert("scale", "0", "0:without scale, 1:per-token/channel scale, only for fp8/bf8") + .insert("scale", "1", "0:without scale, 1:per-token/channel scale, only for fp8/bf8") .insert("persistent", "0", "0: no persistent, 1: persistent kernel") .insert("warp_tile", "0", @@ -421,12 +421,7 @@ int run_flatmm_example(int argc, char* argv[]) int persistent_opt = arg_parser.get_int("persistent"); if(a_layout == "R" && b_layout == "C") { - if(data_type == "fp16") - { - run_flatmm_example_with_layouts>( - argc, argv, Row{}, Col{}, Row{}); - } - else if(data_type == "bf16") + if(data_type == "bf16") { run_flatmm_example_with_layouts>( argc, argv, Row{}, Col{}, Row{}); @@ -435,19 +430,20 @@ int run_flatmm_example(int argc, char* argv[]) { if(scale_opt == 0) { - if(persistent_opt == 0) - { - run_flatmm_example_with_layouts>( - argc, argv, Row{}, Col{}, Row{}); - } - else - { - run_flatmm_example_with_layouts, - -1, - -1, - true>(argc, argv, Row{}, Col{}, Row{}); - } + throw std::runtime_error("scale_opt=0 no enabled to accelerate compiling"); + // if(persistent_opt == 0) + // { + // run_flatmm_example_with_layouts>( + // argc, argv, Row{}, Col{}, Row{}); + // } + // else + // { + // run_flatmm_example_with_layouts, + // -1, + // -1, + // true>(argc, argv, Row{}, Col{}, Row{}); + // } } else { @@ -468,19 +464,6 @@ int run_flatmm_example(int argc, char* argv[]) } } } - else if(data_type == "bf8") - { - if(scale_opt == 0) - { - run_flatmm_example_with_layouts>( - argc, argv, Row{}, Col{}, Row{}); - } - else - { - run_flatmm_example_with_layouts, 1, 1>( - argc, argv, Row{}, Col{}, Row{}); - } - } else { throw std::runtime_error("Unsupported data_type!");