Skip to content

Commit 72e4c21

Browse files
Rob Hessjeffdonahue
authored andcommitted
Remove GPU stubs for all layers with device-unified Forward() and Backward().
1 parent 9a5b6d2 commit 72e4c21

14 files changed

+0
-58
lines changed

src/caffe/layers/conv_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -221,10 +221,6 @@ void ConvolutionLayer<Dtype>::Backward(const vector<Blob<Dtype>*>& top,
221221
}
222222
}
223223

224-
#ifdef CPU_ONLY
225-
STUB_GPU(ConvolutionLayer);
226-
#endif
227-
228224
INSTANTIATE_CLASS(ConvolutionLayer);
229225

230226
} // namespace caffe

src/caffe/layers/eltwise_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -93,10 +93,6 @@ void EltwiseLayer<Dtype>::Backward(const vector<Blob<Dtype>*>& top,
9393
}
9494
}
9595

96-
#ifdef CPU_ONLY
97-
STUB_GPU(EltwiseLayer);
98-
#endif
99-
10096
INSTANTIATE_CLASS(EltwiseLayer);
10197

10298

src/caffe/layers/euclidean_loss_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,10 +48,6 @@ void EuclideanLossLayer<Dtype>::Backward(const vector<Blob<Dtype>*>& top,
4848
}
4949
}
5050

51-
#ifdef CPU_ONLY
52-
STUB_GPU(EuclideanLossLayer);
53-
#endif
54-
5551
INSTANTIATE_CLASS(EuclideanLossLayer);
5652

5753
} // namespace caffe

src/caffe/layers/flatten_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,6 @@ void FlattenLayer<Dtype>::Backward(const vector<Blob<Dtype>*>& top,
3030
(*bottom)[0]->ShareDiff(*top[0]);
3131
}
3232

33-
#ifdef CPU_ONLY
34-
STUB_GPU(FlattenLayer);
35-
#endif
36-
3733
INSTANTIATE_CLASS(FlattenLayer);
3834

3935
} // namespace caffe

src/caffe/layers/hdf5_data_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -113,10 +113,6 @@ Dtype HDF5DataLayer<Dtype>::Forward(const vector<Blob<Dtype>*>& bottom,
113113
return Dtype(0.);
114114
}
115115

116-
#ifdef CPU_ONLY
117-
STUB_GPU_FORWARD(HDF5DataLayer, Forward);
118-
#endif
119-
120116
INSTANTIATE_CLASS(HDF5DataLayer);
121117

122118
} // namespace caffe

src/caffe/layers/hdf5_output_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,10 +62,6 @@ Dtype HDF5OutputLayer<Dtype>::Forward(const vector<Blob<Dtype>*>& bottom,
6262
return Dtype(0.);
6363
}
6464

65-
#ifdef CPU_ONLY
66-
STUB_GPU(HDF5OutputLayer);
67-
#endif
68-
6965
INSTANTIATE_CLASS(HDF5OutputLayer);
7066

7167
} // namespace caffe

src/caffe/layers/im2col_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,6 @@ void Im2colLayer<Dtype>::Backward(const vector<Blob<Dtype>*>& top,
8181
}
8282
}
8383

84-
#ifdef CPU_ONLY
85-
STUB_GPU(Im2colLayer);
86-
#endif
87-
8884
INSTANTIATE_CLASS(Im2colLayer);
8985

9086
} // namespace caffe

src/caffe/layers/image_data_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -265,10 +265,6 @@ Dtype ImageDataLayer<Dtype>::Forward(const vector<Blob<Dtype>*>& bottom,
265265
return Dtype(0.);
266266
}
267267

268-
#ifdef CPU_ONLY
269-
STUB_GPU_FORWARD(ImageDataLayer, Forward);
270-
#endif
271-
272268
INSTANTIATE_CLASS(ImageDataLayer);
273269

274270
} // namespace caffe

src/caffe/layers/inner_product_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,10 +94,6 @@ void InnerProductLayer<Dtype>::Backward(const vector<Blob<Dtype>*>& top,
9494
}
9595
}
9696

97-
#ifdef CPU_ONLY
98-
STUB_GPU(InnerProductLayer);
99-
#endif
100-
10197
INSTANTIATE_CLASS(InnerProductLayer);
10298

10399
} // namespace caffe

src/caffe/layers/power_layer.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,10 +94,6 @@ void PowerLayer<Dtype>::Backward(const vector<Blob<Dtype>*>& top,
9494
}
9595
}
9696

97-
#ifdef CPU_ONLY
98-
STUB_GPU(PowerLayer);
99-
#endif
100-
10197
INSTANTIATE_CLASS(PowerLayer);
10298

10399

0 commit comments

Comments
 (0)