diff --git a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_nwarp_sshuffle_qr_ks_vs.hpp b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_nwarp_sshuffle_qr_ks_vs.hpp index 261fe2c30a..4895c083bb 100644 --- a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_nwarp_sshuffle_qr_ks_vs.hpp +++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_nwarp_sshuffle_qr_ks_vs.hpp @@ -247,7 +247,7 @@ struct BlockFmhaFwdSplitKVPipelineNWarpSShuffleQRKSVS auto o_acc = OaccBlockTileType{}; // infer Sacc, S, P, M, L, Oacc type - using SBlockTileType = decltype(cast_tile(o_acc)); + using SBlockTileType = decltype(cast_tile(s_acc)); using MLBlockTileType = decltype(block_tile_reduce( SBlockTileType{}, sequence<1>{}, f_max, SMPLComputeDataType{0}));