diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp index 05288b0d5d..d705f38d36 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp @@ -91,7 +91,7 @@ struct HstuAttentionFwdPipelineQRKSVS if constexpr(kHasBias) return 2; else - return 3; + return 2; } else if constexpr(kQKHeaddim == 256) {