Avoid too-large block id

360383bb · Po-Yen, Chen · e017a9f0 · 360383bb
Commit 360383bb authored Sep 08, 2022 by Po-Yen, Chen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp +2 -0

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
@@ -54,6 +54,8 @@ struct Block2TileMap
        const auto H0 = math::integer_divide_ceil(desc_.GetLength(Number<NumDim - 2>{}), HPerBlock);
        const auto W0 = math::integer_divide_ceil(desc_.GetLength(Number<NumDim - 1>{}), WPerBlock);

+        block_1d_id = block_1d_id % (H0 * W0);
+
        index_t idx_H0 = block_1d_id / W0;
        index_t idx_W0 = block_1d_id % W0;