"tests/vscode:/vscode.git/clone" did not exist on "6dbef45e6e44dd8c1396c1af30b674abb47ffb66"
Commit 2a16c61c authored by Adam Osewski's avatar Adam Osewski
Browse files

Fix GetNextKTiles.

parent 3f74017c
...@@ -59,26 +59,12 @@ class StridedReductionTileLoop ...@@ -59,26 +59,12 @@ class StridedReductionTileLoop
/// @return The number of next k-tiles to process. /// @return The number of next k-tiles to process.
__device__ index_t GetNextKTiles(index_t k_tiles, index_t k_tile_idx) __device__ index_t GetNextKTiles(index_t k_tiles, index_t k_tile_idx)
{ {
index_t k_tiles_left = k_tiles - k_tile_idx; index_t k_tiles_left = k_tiles - k_tile_idx;
index_t next_k_tiles = index_t block_tiles_left = tiles_per_block_ - block_tile_idx_;
k_tiles_left <= tiles_per_block_ ? k_tiles_left : tiles_per_block_ - block_tile_idx_; index_t next_k_tiles = k_tiles_left <= block_tiles_left ? k_tiles_left : block_tiles_left;
tile_id_ += next_k_tiles; tile_id_ += next_k_tiles;
block_tile_idx_ += next_k_tiles; block_tile_idx_ += next_k_tiles;
if(blockIdx.x < 4 && ck::debug::is_thread_local_1d_id_idx<0>())
{
printf("[GetNextKTiles] bid: %d, k_tiles: %d, k_idx:%d, next_k_tiles: %d, "
"k_tiles_left: %d,"
" tile_id: %d, block_tile_idx: %d\n",
static_cast<index_t>(blockIdx.x),
k_tiles,
k_tile_idx,
next_k_tiles,
k_tiles_left,
tile_id_,
block_tile_idx_);
}
return next_k_tiles; return next_k_tiles;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment