fix shuffle GemmORepeat

762cbddf · danyao12 · 17774771 · 762cbddf
Commit 762cbddf authored May 08, 2023 by danyao12
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

include/ck/tensor_operation/gpu/grid/gridwise_batched_multihead_attention_backward_xdl_cshuffle_pt3.hpp ...batched_multihead_attention_backward_xdl_cshuffle_pt3.hpp +2 -2

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_batched_multihead_attention_backward_xdl_cshuffle_pt3.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_batched_multihead_attention_backward_xdl_cshuffle_pt3.hpp
@@ -2164,7 +2164,7 @@ struct GridwiseBatchedMultiheadAttentionBackward_Xdl_CShuffle_V1
        // shuffle dK&dV and write
        {
            static_assert(Gemm2Params_N_O_M::GemmNRepeat % CShuffleMXdlPerWavePerShuffle == 0 &&
-                              Gemm2Params_N_O_M::GemmNRepeat % CShuffleNXdlPerWavePerShuffle == 0,
+                              Gemm2Params_N_O_M::GemmORepeat % CShuffleNXdlPerWavePerShuffle == 0,
                          "wrong!");

            constexpr index_t MWave = Gemm2Params_N_O_M::GemmNWave;
@@ -2363,7 +2363,7 @@ struct GridwiseBatchedMultiheadAttentionBackward_Xdl_CShuffle_V1

            // space filling curve for threadwise C in VGPR
            constexpr auto sfc_c_vgpr = SpaceFillingCurve<Sequence<Gemm2Params_N_O_M::GemmNRepeat,
-                                                                   Gemm2Params_N_O_M::GemmNRepeat,
+                                                                   Gemm2Params_N_O_M::GemmORepeat,
                                                                   1,
                                                                   1,
                                                                   1,