ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

block index correction

Browse Source
This commit is contained in:
Harisankar Sadasivan
2024-09-11 15:34:27 -04:00
parent 7c63aa5f1c
commit 8b69a9ddd0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_streamk_v3.hpp
View File

@@ -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);