Fix overflow issue in threshold

FolkeV · May 27, 2019 · 1e9ca96 · 1e9ca96
1 parent 8034030
commit 1e9ca96
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/main.cu b/main.cu
@@ -65,7 +65,7 @@ int main(int argc,char **argv){
 	cudaMallocManaged(&d_img   , numPixels * sizeof(char));
 
 	// Pre process image
-	int imgMean = util::mean(image.data, numPixels);
+	unsigned int imgMean = util::mean(image.data, numPixels);
 	util::threshold(d_img, image.data, imgMean, numPixels);
 
 	// Run and time kernel

diff --git a/utils.hpp b/utils.hpp
@@ -26,8 +26,8 @@
 
 namespace util{
 	// Compute mean of image data
-	int mean(const unsigned char* img, const int N){
-		int mean = 0;
+	unsigned int mean(const unsigned char* img, const int N){
+		unsigned int mean = 0;
 
 		for(int i = 0; i<N; i++){
 			mean += img[i];