From 6f9de3e7c5a0979c7d26db2056fff5741c2b374d Mon Sep 17 00:00:00 2001 From: Sergio Date: Tue, 27 May 2014 11:01:19 -0700 Subject: [PATCH] Fixed type in ThresholdLayer.cu --- src/caffe/layers/threshold_layer.cu | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/caffe/layers/threshold_layer.cu b/src/caffe/layers/threshold_layer.cu index 0497507..624a822 100644 --- a/src/caffe/layers/threshold_layer.cu +++ b/src/caffe/layers/threshold_layer.cu @@ -11,7 +11,8 @@ using std::max; namespace caffe { template -__global__ void ThresholdForward(const int n, const, threshold, const Dtype* in, Dtype* out) { +__global__ void ThresholdForward(const int n, const Dtype threshold, + const Dtype* in, Dtype* out) { CUDA_KERNEL_LOOP(index, n) { out[index] = in[index] > threshold ? 1 : 0; } -- 2.7.4