ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 10:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix in GetQKBlockGemm()

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-09-27 14:31:24 +00:00
parent 27b96b15c4
commit 6b40ce4074
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp
View File

@@ -599,9 +599,9 @@ struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy
typename Problem::QKVDataType,
typename Problem::QKVDataType,
typename Problem::GemmAccDataType,
Problem::HstuAttentionTileSetting::Gemm1WarpTile::at(number<0>{}),
Problem::HstuAttentionTileSetting::Gemm1WarpTile::at(number<1>{}),
Problem::HstuAttentionTileSetting::Gemm1WarpTile::at(number<2>{}),
Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<0>{}),
Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<1>{}),
Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<2>{}),
true,
false,
false,
@@ -614,9 +614,9 @@ struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy
typename Problem::QKVDataType,
typename Problem::QKVDataType,
typename Problem::GemmAccDataType,
Problem::HstuAttentionTileSetting::Gemm1WarpTile::at(number<0>{}),
Problem::HstuAttentionTileSetting::Gemm1WarpTile::at(number<1>{}),
Problem::HstuAttentionTileSetting::Gemm1WarpTile::at(number<2>{}),
Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<0>{}),
Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<1>{}),
Problem::HstuAttentionTileSetting::Gemm0WarpTile::at(number<2>{}),
true,
false,
false,