diff --git a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_base.hpp b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_base.hpp index 4973d9c941..dcc11015e7 100644 --- a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_base.hpp +++ b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_base.hpp @@ -237,12 +237,16 @@ struct GemmPipelineAgBgCrImplBase auto a_lds_load_tile_distr = []() { if constexpr(is_a_load_tr) + { return make_static_tile_distribution( typename InputTileDistributionTraits< typename ALdsLoadTileDistr::DstrEncode, typename Problem::ADataType>::TransposedDstrEncode{}); + } else + { return ALdsLoadTileDistr{}; + } }(); auto a_lds_gemm_window = @@ -320,12 +324,15 @@ struct GemmPipelineAgBgCrImplBase auto b_lds_load_tile_distr = []() { if constexpr(is_b_load_tr) + { return make_static_tile_distribution( typename InputTileDistributionTraits::TransposedDstrEncode{}); - + } else + { return BLdsLoadTileDistr{}; + } }(); auto b_lds_gemm_window =