From f634899f44fda89692e12b8114889838f50a73d8 Mon Sep 17 00:00:00 2001 From: Kai Li Date: Tue, 25 Feb 2014 19:21:18 +0800 Subject: [PATCH] Instantiate caffe_cpu_sign for float and double --- src/caffe/util/math_functions.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/caffe/util/math_functions.cpp b/src/caffe/util/math_functions.cpp index 04a5228..bf24488 100644 --- a/src/caffe/util/math_functions.cpp +++ b/src/caffe/util/math_functions.cpp @@ -410,4 +410,10 @@ void caffe_gpu_asum(const int n, const double* x, double* y) { CUBLAS_CHECK(cublasDasum(Caffe::cublas_handle(), n, x, 1, y)); } +template <> +void caffe_cpu_sign(const int n, const float* x, float* y); + +template <> +void caffe_cpu_sign(const int n, const double* x, double* y); + } // namespace caffe -- 2.7.4