Avoid address overflow (winner_takes_all)

fixstars · Aug 21, 2023 · 99be2de · 99be2de
1 parent 401f27b
commit 99be2de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/winner_takes_all.cu b/src/winner_takes_all.cu
@@ -91,7 +91,7 @@ __global__ void winner_takes_all_kernel(
 			? REDUCTION_PER_THREAD
 			: ACCUMULATION_INTERVAL;
 
-	const unsigned int cost_step = MAX_DISPARITY * width * height;
+	const size_t cost_step = static_cast<size_t>(MAX_DISPARITY) * width * height;
 	const unsigned int warp_id = threadIdx.x / WARP_SIZE;
 	const unsigned int lane_id = threadIdx.x % WARP_SIZE;