updated a bunch of things, ready to test if it breaks things
[jacinto-ai/caffe-jacinto.git] / src / caffe / util / math_functions.hpp
index 822ef31be74acea4ed4a53851229c67bf32fbfed..f09afe38eedfc2a1223ce6fc57af2dc45eddd4df 100644 (file)
@@ -45,6 +45,9 @@ void caffe_copy(const int N, const Dtype *X, Dtype *Y);
 template <typename Dtype>
 void caffe_scal(const int N, const Dtype alpha, Dtype *X);
 
+template <typename Dtype>
+void caffe_gpu_scal(const int N, const Dtype alpha, Dtype *X);
+
 template <typename Dtype>
 void caffe_sqr(const int N, const Dtype* a, Dtype* y);