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

Tiny change in pipeline BlockGemm definition to adapt to the latest merging with develop branch

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-08-18 14:51:26 +00:00
parent 89cd5ff5fe
commit 7b68b6eebb
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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,