From 8cb4e45fb7301b02e36df40827f8fc56eaf8e2f6 Mon Sep 17 00:00:00 2001 From: Sergio Date: Tue, 27 May 2014 11:02:20 -0700 Subject: [PATCH] Fixed call to ThresholdForward in ThresholdLayer.cu --- src/caffe/layers/threshold_layer.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/caffe/layers/threshold_layer.cu b/src/caffe/layers/threshold_layer.cu index 624a822..61b1eca 100644 --- a/src/caffe/layers/threshold_layer.cu +++ b/src/caffe/layers/threshold_layer.cu @@ -25,7 +25,7 @@ Dtype ThresholdLayer::Forward_gpu(const vector*>& bottom, Dtype* top_data = (*top)[0]->mutable_gpu_data(); const int count = bottom[0]->count(); // NOLINT_NEXT_LINE(whitespace/operators) - ReLUForward<<>>( + ThresholdForward<<>>( count, threshold_, bottom_data, top_data); CUDA_POST_KERNEL_CHECK; -- 2.7.4