block index correction

8b69a9dd · Harisankar Sadasivan · 7c63aa5f · 8b69a9dd
Commit 8b69a9dd authored Sep 11, 2024 by Harisankar Sadasivan
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_streamk_v3.hpp ...ration/gpu/grid/gridwise_gemm_xdl_cshuffle_streamk_v3.hpp +1 -1

No files found.
--- 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);