Merge pull request BVLC#2016 from shelhamer/share-col-buffer

Share convolution buffers to reduce memory usage
elleryrussell · Apr 24, 2015 · 12d29cd · 12d29cd
2 parents 2dd61dc + 1dcfc3c
commit 12d29cd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp
@@ -107,7 +107,7 @@ class BaseConvolutionLayer : public Layer<Dtype> {
  int col_offset_;
  int output_offset_;
 
- Blob<Dtype> col_buffer_;
+ static Blob<Dtype> col_buffer_;
  Blob<Dtype> bias_multiplier_;
 };
 

diff --git a/src/caffe/layers/base_conv_layer.cpp b/src/caffe/layers/base_conv_layer.cpp
@@ -8,6 +8,9 @@
 
 namespace caffe {
 
+template <typename Dtype>
+Blob<Dtype> BaseConvolutionLayer<Dtype>::col_buffer_;
+
 template <typename Dtype>
 void BaseConvolutionLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype>*>& bottom,
  const vector<Blob<Dtype>*>& top) {