fix dq gemm a_origin

9879bcd5 · danyao12 · b03f56db · 9879bcd5
Commit 9879bcd5 authored Feb 07, 2023 by danyao12
Hide 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_v2.hpp ..._batched_multihead_attention_backward_xdl_cshuffle_v2.hpp +2 -2

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_batched_multihead_attention_backward_xdl_cshuffle_v2.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_batched_multihead_attention_backward_xdl_cshuffle_v2.hpp
@@ -1333,8 +1333,8 @@ struct GridwiseBatchedMultiheadAttentionBackward_Xdl_CShuffle
            typename Gemm1::ABlockwiseCopy{tensor_operation::element_wise::PassThrough{}};
        // dQ: blockwise gemm
-        auto qgrad_blockwise_gemm = typename Gemm1::BlockwiseGemm{};
+        auto qgrad_blockwise_gemm =
-        qgrad_blockwise_gemm.SetBBlockStartWindow(make_tuple(0, 0, 0, 0));
+            typename Gemm1::BlockwiseGemm{make_tuple(0, 0, 0, 0), make_tuple(0, 0, 0, 0)};
        // dQ: B matrix blockwise copy        
        auto k_thread_origin = qgrad_blockwise_gemm.CalculateBThreadOriginDataIndex();