src/caffe/util/math_functions.hpp

   1 // Copyright 2013 Yangqing Jia
   2
   3 #ifndef CAFFE_UTIL_MATH_FUNCTIONS_H_
   4 #define CAFFE_UTIL_MATH_FUNCTIONS_H_
   5
   6 #include <mkl.h>
   7 #include <cublas_v2.h>
   8
   9 namespace caffe {
  10
  11 // Decaf gemm provides a simpler interface to the gemm functions, with the
  12 // limitation that the data has to be contiguous in memory.
  13 template <typename Dtype>
  14 void caffe_cpu_gemm(const CBLAS_TRANSPOSE TransA,
  15     const CBLAS_TRANSPOSE TransB, const int M, const int N, const int K,
  16     const Dtype alpha, const Dtype* A, const Dtype* B, const Dtype beta,
  17     Dtype* C);
  18
  19 // Decaf gpu gemm provides an interface that is almost the same as the cpu
  20 // gemm function - following the c convention and calling the fortran-order
  21 // gpu code under the hood.
  22 template <typename Dtype>
  23 void caffe_gpu_gemm(const CBLAS_TRANSPOSE TransA,
  24     const CBLAS_TRANSPOSE TransB, const int M, const int N, const int K,
  25     const Dtype alpha, const Dtype* A, const Dtype* B, const Dtype beta,
  26     Dtype* C);
  27
  28 template <typename Dtype>
  29 void caffe_cpu_gemv(const CBLAS_TRANSPOSE TransA, const int M, const int N,
  30     const Dtype alpha, const Dtype* A, const Dtype* x, const Dtype beta,
  31     Dtype* y);
  32
  33 template <typename Dtype>
  34 void caffe_gpu_gemv(const CBLAS_TRANSPOSE TransA, const int M, const int N,
  35     const Dtype alpha, const Dtype* A, const Dtype* x, const Dtype beta,
  36     Dtype* y);
  37
  38 template <typename Dtype>
  39 void caffe_axpy(const int N, const Dtype alpha, const Dtype* X,
  40     Dtype* Y);
  41
  42 template <typename Dtype>
  43 void caffe_axpby(const int N, const Dtype alpha, const Dtype* X,
  44     const Dtype beta, Dtype* Y);
  45
  46 template <typename Dtype>
  47 void caffe_copy(const int N, const Dtype *X, Dtype *Y);
  48
  49 template <typename Dtype>
  50 void caffe_scal(const int N, const Dtype alpha, Dtype *X);
  51
  52 template <typename Dtype>
  53 void caffe_gpu_scal(const int N, const Dtype alpha, Dtype *X);
  54
  55 template <typename Dtype>
  56 void caffe_sqr(const int N, const Dtype* a, Dtype* y);
  57
  58 template <typename Dtype>
  59 void caffe_add(const int N, const Dtype* a, const Dtype* b, Dtype* y);
  60
  61 template <typename Dtype>
  62 void caffe_sub(const int N, const Dtype* a, const Dtype* b, Dtype* y);
  63
  64 template <typename Dtype>
  65 void caffe_mul(const int N, const Dtype* a, const Dtype* b, Dtype* y);
  66
  67 template <typename Dtype>
  68 void caffe_div(const int N, const Dtype* a, const Dtype* b, Dtype* y);
  69
  70 template <typename Dtype>
  71 void caffe_powx(const int n, const Dtype* a, const Dtype b, Dtype* y);
  72
  73 template <typename Dtype>
  74 void caffe_vRngUniform(const int n, Dtype* r, const Dtype a, const Dtype b);
  75
  76 template <typename Dtype>
  77 void caffe_vRngGaussian(const int n, Dtype* r, const Dtype a,
  78     const Dtype sigma);
  79
  80 template <typename Dtype>
  81 void caffe_exp(const int n, const Dtype* a, Dtype* y);
  82
  83 template <typename Dtype>
  84 Dtype caffe_cpu_dot(const int n, const Dtype* x, const Dtype* y);
  85
  86 }  // namespace caffe
  87
  88
  89 #endif  // CAFFE_UTIL_MATH_FUNCTIONS_H_