check lef bottom corner for tile skipping

616a51fd · wangshaojie6 · 878f66fb · 616a51fd
Commit 616a51fd authored Sep 19, 2022 by wangshaojie6
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

include/ck/tensor_operation/gpu/grid/gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp ...id/gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp +1 -2

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp
@@ -772,8 +772,7 @@ struct GridwiseBatchedGemmSoftmaxGemm_Xdl_CShuffle
                auto gemm0_n_block_idx =
                    __builtin_amdgcn_readfirstlane(gemm1_k_block_outer_index * NPerBlock);
                if((m_block_data_idx_on_grid < gemm0_n_block_idx) &&
-                   ((m_block_data_idx_on_grid + MPerBlock - 1) <
+                   (m_block_data_idx_on_grid < (gemm0_n_block_idx + NPerBlock - 1)))
-                    (gemm0_n_block_idx + NPerBlock - 1)))
                {
                    continue;
                }