Fix aligment for o_acc

12871dd4 · Po Yen Chen · 9bf87a90 · 12871dd4
Commit 12871dd4 authored Dec 29, 2024 by Po Yen Chen
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 ...peline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs_async.hpp +1 -3

No files found.
--- a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs_async.hpp
+++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs_async.hpp
@@ -71,9 +71,7 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVSAsync
                   : kPadSeqLenK ? 1
                                 : Policy::template GetAlignmentV<Problem>();
    }();
-    static constexpr index_t kAlignmentOacc =
-        kPadHeadDimV ? 1 : Policy::template GetAlignmentOacc<Problem>();
-
+    static constexpr index_t kAlignmentOacc = Policy::template GetAlignmentOacc<Problem>();
    static constexpr index_t kAlignmentBias =
        kPadSeqLenK ? 1 : Policy::template GetAlignmentBias<Problem>();