From 8b69a9ddd05b4c8a7bad27f822b30b92cd236202 Mon Sep 17 00:00:00 2001 From: Harisankar Sadasivan Date: Wed, 11 Sep 2024 15:34:27 -0400 Subject: [PATCH] block index correction --- .../gpu/grid/gridwise_gemm_xdl_cshuffle_streamk_v3.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_streamk_v3.hpp b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_streamk_v3.hpp index b17a816ad5..3aa1aada7d 100644 --- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_streamk_v3.hpp +++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_streamk_v3.hpp @@ -1303,7 +1303,7 @@ struct GridwiseGemm_xdl_cshuffle_streamk_v3 // start to compute auto reduction_idx = - blockIdx.x - block_2_ctile_map_streamk.reduction_start_block_idx; + block_idx - block_2_ctile_map_streamk.reduction_start_block_idx; auto spatial_idx = block_2_ctile_map_streamk.tile_to_spatial( reduction_idx, problem.M, problem.N);