check for cpu_only build when using caffe_gpu_dot in layer.cpp

lopho · Aug 20, 2016 · ac57884 · ac57884
1 parent 9a8236e
commit ac57884
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/caffe/layer.cpp b/src/caffe/layer.cpp
@@ -24,7 +24,9 @@ void Layer<Dtype>::Unlock() {
 
 template <typename Dtype>
 void Layer<Dtype>::layer_gpu_dot(const int n, const Dtype* x, const Dtype* y, Dtype* out) {
+#ifndef CPU_ONLY
   caffe_gpu_dot(n, x, y, out);
+#endif
 }