evquantize: unsigned saturate range fix.

vera121 · vera121 · commit ec65665cd8ce · 2018-12-13T23:44:23.000-05:00
diff --git a/src/caffe/util/math_functions.cu b/src/caffe/util/math_functions.cu
@@ -229,6 +229,8 @@ __global__ void unsigned_saturate_kernel(const int n, Dtype* y) {
   CUDA_KERNEL_LOOP(index, n) {
     if(y[index] > UNSIGNED_SATURATE_MAX)
       y[index] = SIGNED_SATURATE_MAX;
+    if(y[index] < 0)
+      y[index] = 0;
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -229,6 +229,8 @@ __global__ void unsigned_saturate_kernel(const int n, Dtype* y) {`
`229`	`229`	`CUDA_KERNEL_LOOP(index, n) {`
`230`	`230`	`if(y[index] > UNSIGNED_SATURATE_MAX)`
`231`	`231`	`y[index] = SIGNED_SATURATE_MAX;`
	`232`	`+ if(y[index] < 0)`
	`233`	`+ y[index] = 0;`
`232`	`234`	`}`
`233`	`235`	`}`
`234`	`236`