modify comment

9438a118 · danyao12 · 1a306e0d · 9438a118
Commit 9438a118 authored May 24, 2023 by danyao12
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

include/ck/tensor_operation/gpu/grid/gridwise_batched_multihead_attention_backward_xdl_cshuffle_pt4.hpp ...batched_multihead_attention_backward_xdl_cshuffle_pt4.hpp +3 -3

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_batched_multihead_attention_backward_xdl_cshuffle_pt4.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_batched_multihead_attention_backward_xdl_cshuffle_pt4.hpp
@@ -1285,8 +1285,8 @@ struct GridwiseBatchedMultiheadAttentionBackward_Xdl_CShuffle_V1

        // 6 GEMM operations are categorized into 3 buckets. SizeK == SizeO == head_dim
        // S_MNK  / dP_MNO Gemm (Gemm0 rcr)
-        // Y_MON / dQ_MKN Gemm (Gemm1 rrr)
-        // dV_NOM / dK_NKM Gemm (Gemm2 crr)
+        // dV_NOM / dK_NKM Gemm (Gemm1 rrr)
+        // Y_MON  / dQ_MKN Gemm (Gemm2 crr)

        // LDS allocation for Q / K / V / dY
        auto q_block_buf = make_dynamic_buffer<AddressSpaceEnum::Lds>(