From c1817464bef300f6f0e8841c7aa04c09b6fcbf73 Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Sun, 30 Nov 2025 14:04:48 +0000 Subject: [PATCH] Tiny fix in GetQKBlockGemm --- .../hstu_attention_fwd_pipeline_default_policy.hpp | 12 ------------ 1 file changed, 12 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 1d942089f7..f54f3ac0c9 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 @@ -738,18 +738,6 @@ struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy static_assert((WarpGemmM == 16 && WarpGemmK == 32) || (WarpGemmM == 32 && WarpGemmK == 16), "Not supported WarpGemm sizes!"); - - return WarpGemmDispatcher< - typename Problem::QKVDataType, - typename Problem::QKVDataType, - typename Problem::GemmAccDataType, - Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<0>{}), - Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<1>{}), - Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<2>{}), - true, - false, - false, - WGAttrNumAccessEnum::Single>{}; #else static_assert((WarpGemmM == 16 && (WarpGemmK == 16 || WarpGemmK == 32)) || (WarpGemmM == 32 && (WarpGemmK == 8 || WarpGemmK == 16)),