From aea72fabf7e136d3d6eb5d80ce3e4d4a34fba88f Mon Sep 17 00:00:00 2001 From: pnikolic-amd Date: Thu, 12 Jun 2025 11:43:15 -0400 Subject: [PATCH] make changes so that ck_tile gemm example runs on ctr-navi4x-aj50-ws02 / Navi4x --- example/ck_tile/03_gemm/gemm_basic.cpp | 18 +++++++++++------- example/ck_tile/03_gemm/run_gemm_example.inc | 3 ++- 2 files changed, 13 insertions(+), 8 deletions(-) diff --git a/example/ck_tile/03_gemm/gemm_basic.cpp b/example/ck_tile/03_gemm/gemm_basic.cpp index de9608bcb4..cf0c26e1d2 100644 --- a/example/ck_tile/03_gemm/gemm_basic.cpp +++ b/example/ck_tile/03_gemm/gemm_basic.cpp @@ -186,7 +186,8 @@ int run_gemm_example(int argc, char* argv[]) if(data_type == "fp16") { - return run_gemm_example_prec_type(a_layout, b_layout, argc, argv); +// return run_gemm_example_prec_type(a_layout, b_layout, argc, argv); + throw std::runtime_error("Unsupported data type for this operation !!!"); } else if(data_type == "bf16") { @@ -194,21 +195,24 @@ int run_gemm_example(int argc, char* argv[]) } else if(data_type == "fp8") { - return run_gemm_example_prec_type( - a_layout, b_layout, argc, argv); + // return run_gemm_example_prec_type( + // a_layout, b_layout, argc, argv); + throw std::runtime_error("Unsupported data type for this operation !!!"); } else if(data_type == "bf8") { - return run_gemm_example_prec_type( - a_layout, b_layout, argc, argv); + // return run_gemm_example_prec_type( + // a_layout, b_layout, argc, argv); + throw std::runtime_error("Unsupported data type for this operation !!!"); } #if(CK_TILE_PIPELINE_DEFAULT == CK_TILE_PIPELINE_COMPUTE_V3) else if(data_type == "pk_int4_t") { // TODO: Add support for bhalf_t ADataType - return run_gemm_example_prec_type( - a_layout, b_layout, argc, argv); + // return run_gemm_example_prec_type( + // a_layout, b_layout, argc, argv); + throw std::runtime_error("Unsupported data type for this operation !!!"); } #endif else diff --git a/example/ck_tile/03_gemm/run_gemm_example.inc b/example/ck_tile/03_gemm/run_gemm_example.inc index bf455a6415..dd2c2258f1 100644 --- a/example/ck_tile/03_gemm/run_gemm_example.inc +++ b/example/ck_tile/03_gemm/run_gemm_example.inc @@ -248,7 +248,8 @@ int run_gemm_example_with_layouts(int argc, ck_tile::index_t stride_B = arg_parser.get_int("stride_b"); ck_tile::index_t stride_C = arg_parser.get_int("stride_c"); - ck_tile::index_t kbatch = arg_parser.get_int("split_k"); + // ck_tile::index_t kbatch = arg_parser.get_int("split_k"); + ck_tile::index_t kbatch = 1; int n_warmup = arg_parser.get_int("warmup"); int n_repeat = arg_parser.get_int("repeat"); ck_tile::index_t init_method = arg_parser.get_int("init");