diff options
Diffstat (limited to 'src/caffe/common.hpp')
| -rw-r--r-- | src/caffe/common.hpp | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/src/caffe/common.hpp b/src/caffe/common.hpp index 39e417f7..8eb79876 100644 --- a/src/caffe/common.hpp +++ b/src/caffe/common.hpp @@ -43,18 +43,14 @@ private:\ namespace caffe { -// Two classes whose purpose are solely for instantiating blob template -// functions. -class GPUBrewer {}; -class CPUBrewer {}; // We will use the boost shared_ptr instead of the new C++11 one mainly // because cuda does not work (at least now) well with C++11 features. using boost::shared_ptr; -// For backward compatibility we will just use 512 threads per block -const int CAFFE_CUDA_NUM_THREADS = 512; +// We will use 1024 threads per block, which requires cuda sm_2x or above. +const int CAFFE_CUDA_NUM_THREADS = 1024; inline int CAFFE_GET_BLOCKS(const int N) { |