Fix CPU only

ctensmeyer · Jun 23, 2017 · 4166903 · 4166903
1 parent 61fdd11
commit 4166903
Show file tree

Hide file tree

Showing 5 changed files with 0 additions and 20 deletions.
diff --git a/src/caffe/layers/graph4c_construction_layer.cpp b/src/caffe/layers/graph4c_construction_layer.cpp
@@ -229,10 +229,6 @@ void Graph4CConstructorLayer<Dtype>::Forward_cpu(
 }
 
 
-#ifdef CPU_ONLY
-STUB_GPU_FORWARD(Graph4CConstructorLayer);
-#endif
-
 INSTANTIATE_CLASS(Graph4CConstructorLayer);
 REGISTER_LAYER_CLASS(Graph4CConstructor);
 

diff --git a/src/caffe/layers/graph4c_energy_layer.cpp b/src/caffe/layers/graph4c_energy_layer.cpp
@@ -251,10 +251,6 @@ void Graph4CEnergyLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& top,
 }
 
 
-#ifdef CPU_ONLY
-STUB_GPU(Graph4CEnergyLayer);
-#endif
-
 INSTANTIATE_CLASS(Graph4CEnergyLayer);
 REGISTER_LAYER_CLASS(Graph4CEnergy);
 

diff --git a/src/caffe/layers/graph4c_make_modular_layer.cpp b/src/caffe/layers/graph4c_make_modular_layer.cpp
@@ -173,10 +173,6 @@ void Graph4CMakeModularLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& to
 }
 
 
-#ifdef CPU_ONLY
-STUB_GPU(Graph4CMakeModularLayer);
-#endif
-
 INSTANTIATE_CLASS(Graph4CMakeModularLayer);
 REGISTER_LAYER_CLASS(Graph4CMakeModular);
 

diff --git a/src/caffe/layers/graph4c_pair_layer.cpp b/src/caffe/layers/graph4c_pair_layer.cpp
@@ -99,10 +99,6 @@ void Graph4CPairLayer<Dtype>::Forward_cpu(
 }
 
 
-#ifdef CPU_ONLY
-STUB_GPU_FORWARD(Graph4CPairLayer);
-#endif
-
 INSTANTIATE_CLASS(Graph4CPairLayer);
 REGISTER_LAYER_CLASS(Graph4CPair);
 

diff --git a/src/caffe/layers/relative_darkness.cpp b/src/caffe/layers/relative_darkness.cpp
@@ -246,10 +246,6 @@ void RelativeDarknessLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& top,
 }
 
 
-#ifdef CPU_ONLY
-STUB_GPU(RelativeDarknessLayer);
-#endif
-
 INSTANTIATE_CLASS(RelativeDarknessLayer);
 REGISTER_LAYER_CLASS(RelativeDarkness);