From ee33d3861ba4cf3c10bd979644cbd89709d3df4e Mon Sep 17 00:00:00 2001
From: Stefano Pigozzi <me@steffo.eu>
Date: Tue, 29 Nov 2022 17:24:48 +0100
Subject: [PATCH] Use `unsigned int` as indexes to make overflows harder

---
 atax/atax.cu | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/atax/atax.cu b/atax/atax.cu
index 88a2524..349b234 100644
--- a/atax/atax.cu
+++ b/atax/atax.cu
@@ -28,7 +28,7 @@
 __host__ static void init_array(DATA_TYPE** A, DATA_TYPE* X)
 {
 	/* X = [ 3.14, 6.28, 9.42, ... ] */
-	for (int y = 0; y < NY; y++) 
+	for (unsigned int y = 0; y < NY; y++) 
 	{
 		X[y] = y * M_PI;
 	}
@@ -42,11 +42,11 @@ __host__ static void init_array(DATA_TYPE** A, DATA_TYPE* X)
 	 *		...
 	 *	]
 	 */
-	for (int x = 0; x < NX; x++) 
+	for (unsigned int x = 0; x < NX; x++) 
 	{
-		for (int y = 0; y < NY; y++) 
+		for (unsigned int y = 0; y < NY; y++) 
 		{
-			A[x][y] = ((DATA_TYPE)x * (y + 1)) / NX;
+			A[x][y] = (DATA_TYPE)(x * (y + 1)) / NX;
 		}
 	}
 }
@@ -61,7 +61,7 @@ __host__ static void init_array(DATA_TYPE** A, DATA_TYPE* X)
  */
 __host__ static void print_array(DATA_TYPE* Y)
 {
-	for (int x = 0; x < NX; x++) 
+	for (unsigned int x = 0; x < NX; x++) 
 	{
 		fprintf(stderr, DATA_PRINTF_MODIFIER, Y[x]);
 	}
@@ -78,21 +78,21 @@ __host__ static void print_array(DATA_TYPE* Y)
  */
 __host__ static void kernel_atax(DATA_TYPE** A, DATA_TYPE* X, DATA_TYPE* Y)
 {
-	for (int x = 0; x < NY; x++) 
+	for (unsigned int x = 0; x < NY; x++) 
 	{
 		Y[i] = 0;
 	}
 	
-	for (int i = 0; i < NX; i++) 
+	for (unsigned int i = 0; i < NX; i++) 
 	{
 		DATA_TYPE tmp = 0;
 		
-		for (int j = 0; j < NY; j++) 
+		for (unsigned int j = 0; j < NY; j++) 
 		{
 			tmp += A[i][j] * X[j];
 		}
 		
-		for (int j = 0; j < NY; j++) 
+		for (unsigned int j = 0; j < NY; j++) 
 		{
 			Y[j] = Y[j] + A[i][j] * tmp;
 		}