From bf69235cfbf7c28ec18caad7d5da80ba98c5ca40 Mon Sep 17 00:00:00 2001 From: root Date: Wed, 26 Mar 2025 13:33:48 +0000 Subject: [PATCH] [GEMM] modify if-else locations --- example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp b/example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp index 4e15ff2845..8b6b845577 100644 --- a/example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp +++ b/example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp @@ -65,11 +65,6 @@ struct GridGemm constexpr auto block_gemm_pipeline = Policy::template GetBlockGemmPipeline(); __shared__ char p_smem_char[block_gemm_pipeline.GetStaticLdsSize()]; - - const auto acc_block_tile = block_gemm_pipeline(a_block_window, - b_block_window, - K / kKPerBlock, - p_smem_char); #else #pragma message ("enable instruction scheduling") // Block GEMM pipeline w/ instruction scheduling @@ -121,12 +116,11 @@ struct GridGemm constexpr auto block_gemm_pipeline = GemmPipelineAgBgCrCompV3(); __shared__ char p_smem_char[block_gemm_pipeline.GetSmemSize()]; - +#endif const auto acc_block_tile = block_gemm_pipeline(a_block_window, b_block_window, K / kKPerBlock, p_smem_char); -#endif // cast to CDataType and apply CElementFunction const auto c_block_tile = tile_elementwise_in(