updated a bunch of things, ready to test if it breaks things

[jacinto-ai/caffe-jacinto.git] / src / caffe / util / math_functions.hpp

diff --git a/src/caffe/util/math_functions.hpp b/src/caffe/util/math_functions.hpp
index 0c03c5929ec185d3328735cde864b6ad9a81e206..f09afe38eedfc2a1223ce6fc57af2dc45eddd4df 100644 (file)
--- a/src/caffe/util/math_functions.hpp
+++ b/src/caffe/util/math_functions.hpp
@@ -43,7 +43,10 @@ template <typename Dtype>
 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);
+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);
@@ -64,6 +67,12 @@ template <typename Dtype>
 void caffe_vRngGaussian(const int n, Dtype* r, const Dtype a,
     const Dtype sigma);
 
+template <typename Dtype>
+void caffe_exp(const int n, const Dtype* a, Dtype* y);
+
+template <typename Dtype>
+Dtype caffe_cpu_dot(const int n, const Dtype* x, const Dtype* y);
+
 }  // namespace caffe