From 863383bb7977a9362018fe925871dc1fa3c714b8 Mon Sep 17 00:00:00 2001
From: Tobias Ribizel <ribizel@kit.edu>
Date: Tue, 17 Dec 2019 13:45:02 +0100
Subject: [PATCH] fix overflow in thread index calculation

Co-Authored-By: Yuhsiang M. Tsai <yhmtsai@gmail.com>
---
 common/matrix/csr_kernels.hpp.inc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/common/matrix/csr_kernels.hpp.inc b/common/matrix/csr_kernels.hpp.inc
index 2e03ca60678..a7fb5654b62 100644
--- a/common/matrix/csr_kernels.hpp.inc
+++ b/common/matrix/csr_kernels.hpp.inc
@@ -501,7 +501,7 @@ __global__ __launch_bounds__(default_block_size) void spgeam_nnz(
     IndexType num_rows, IndexType *nnz)
 {
     constexpr auto sentinel = device_numeric_limits<IndexType>::max;
-    auto row = threadIdx.x + blockDim.x * blockIdx.x;
+    auto row = threadIdx.x + blockDim.x * static_cast<size_type>(blockIdx.x);
     if (row >= num_rows) {
         return;
     }
@@ -544,7 +544,7 @@ __global__ __launch_bounds__(default_block_size) void spgeam(
     const IndexType *c_row_ptrs, IndexType *c_col_idxs, ValueType *c_vals)
 {
     constexpr auto sentinel = device_numeric_limits<IndexType>::max;
-    auto row = threadIdx.x + blockDim.x * blockIdx.x;
+    auto row = threadIdx.x + blockDim.x * static_cast<size_type>(blockIdx.x);
     if (row >= num_rows) {
         return;
     }