Skip to content

Commit

Permalink
Use size_t for loop index in map_encode
Browse files Browse the repository at this point in the history
  • Loading branch information
miscco committed Nov 10, 2023
1 parent e2f7d5d commit f6e5462
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -231,7 +231,7 @@ CUtensorMap map_encode(T *tensor_ptr, Gmem_Dims gmem_dims, Smem_Dims smem_dims)
cuda::std::array<uint64_t, rank-1> stride;
uint64_t base_stride = sizeof(T);

for (int i = 0; i < rank-1; ++i) {
for (size_t i = 0; i < rank-1; ++i) {
base_stride *= gmem_dims[i];
stride[i] = base_stride;
}
Expand Down

0 comments on commit f6e5462

Please sign in to comment.