diff --git a/example/ck_tile/17_grouped_gemm/grouped_gemm.cpp b/example/ck_tile/17_grouped_gemm/grouped_gemm.cpp index 6342bc73e2..4d88f5edec 100644 --- a/example/ck_tile/17_grouped_gemm/grouped_gemm.cpp +++ b/example/ck_tile/17_grouped_gemm/grouped_gemm.cpp @@ -271,7 +271,14 @@ int run_gemm_example_prec_type(std::string a_layout, std::string b_layout, std:: CDataType, AccDataType>(argc, argv, Row{}, Col{}, Col{}); } - + else if(a_layout == "C" && b_layout == "R" && c_layout == "C") + { + return run_grouped_gemm_example_with_layouts(argc, argv, Col{}, Row{}, Col{}); + } else { throw std::runtime_error("Unsupported data layout configuration for A, B and C tensors!");