From 7b68b6eebbd0cf4db40484e4ef41ba4e1aa9c5fc Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Mon, 18 Aug 2025 14:51:26 +0000 Subject: [PATCH] Tiny change in pipeline BlockGemm definition to adapt to the latest merging with develop branch --- .../hstu_attention_fwd_pipeline_default_policy.hpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp index b4349a4032..6a01e5daba 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp @@ -701,7 +701,7 @@ struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy else if constexpr(WarpGemmM == 16) { if constexpr(WarpGemmK == 32) - return WarpGemmMfmaF16F16F32M16N16K32TransposedCDistribution{}; + return WarpGemmMfmaF16F16F32M16N16K32TransposedCDistribution<>{}; else return WarpGemmMfmaF16F16F32M16N16K16TransposedCDistribution{}; } @@ -716,7 +716,7 @@ struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy else if constexpr(WarpGemmM == 16) { if constexpr(WarpGemmK == 32) - return WarpGemmMfmaBf16Bf16F32M16N16K32TransposedCDistribution{}; + return WarpGemmMfmaBf16Bf16F32M16N16K32TransposedCDistribution<>{}; else return WarpGemmMfmaBf16Bf16F32M16N16K16TransposedCDistribution{}; } @@ -781,7 +781,7 @@ struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy } else { - return WarpGemmMfmaDispatcher< + return WarpGemmDispatcher< typename Problem::QKVDataType, typename Problem::QKVDataType, typename Problem::GemmAccDataType,