From 88601c8a0547ee27b73e2c7fafb988bd1bd8e582 Mon Sep 17 00:00:00 2001 From: joye Date: Wed, 21 May 2025 13:31:13 +0800 Subject: [PATCH] update async policy --- .../ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async.hpp | 3 --- .../gemm_pipeline_ag_bg_cr_comp_async_default_policy.hpp | 2 ++ 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async.hpp b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async.hpp index e14591688b..8a0ec47934 100644 --- a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async.hpp +++ b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async.hpp @@ -99,9 +99,6 @@ struct GemmPipelineAgBgCrCompAsync : public BaseGemmPipelineAgBgCrCompAsync(); diff --git a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async_default_policy.hpp b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async_default_policy.hpp index 0957c88967..7a27702966 100644 --- a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async_default_policy.hpp +++ b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async_default_policy.hpp @@ -17,6 +17,8 @@ namespace ck_tile { struct GemmPipelineAgBgCrCompAsyncDefaultPolicy : public UniversalGemmBasePolicy { + static constexpr auto ATileAccessPattern = tile_distribution_pattern::warp_raked; + static constexpr auto BTileAccessPattern = tile_distribution_pattern::warp_raked; template CK_TILE_HOST_DEVICE static constexpr auto MakeALdsBlockDescriptor()