ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 03:37:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

[GEMM] modify if-else locations

Browse Source
This commit is contained in:
root
2025-03-26 13:33:48 +00:00
committed by Philip Maybank
parent ba8b5112c4
commit bf69235cfb
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/ck_tile/99_toy_example/02_gemm/grid_gemm.hpp
View File

@@ -65,11 +65,6 @@ struct GridGemm
constexpr auto block_gemm_pipeline = Policy::template GetBlockGemmPipeline<Problem>();
__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<UniversalGemmProblem>();
__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(