Skip to content

GitLab

Commit 12871dd4 authored by Po Yen Chen's avatar Po Yen Chen
Browse files

Fix aligment for o_acc

parent 9bf87a90
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs_async.hpp
View file @ 12871dd4
...@@ -71,9 +71,7 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVSAsync ...@@ -71,9 +71,7 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVSAsync
: kPadSeqLenK ? 1 : kPadSeqLenK ? 1
: Policy::template GetAlignmentV<Problem>(); : Policy::template GetAlignmentV<Problem>();
}(); }();
static constexpr index_t kAlignmentOacc = static constexpr index_t kAlignmentOacc = Policy::template GetAlignmentOacc<Problem>();
kPadHeadDimV ? 1 : Policy::template GetAlignmentOacc<Problem>();
static constexpr index_t kAlignmentBias = static constexpr index_t kAlignmentBias =
kPadSeqLenK ? 1 : Policy::template GetAlignmentBias<Problem>(); kPadSeqLenK ? 1 : Policy::template GetAlignmentBias<Problem>();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment