Merge pull request BVLC#149 from drnikolaev/caffe-0.15-fix

Setting maximum roundable bin size to 4M
lukeyeager · May 27, 2016 · 2f0fbed · 2f0fbed
2 parents 72c81ba + 22ef291
commit 2f0fbed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/util/gpu_memory.cpp b/src/caffe/util/gpu_memory.cpp
@@ -142,7 +142,7 @@ void GPUMemoryManager::InitMemory(const std::vector<int>& gpus, PoolMode m) {
     try {
       // Just in case someone installed 'no cleanup' arena before
       delete cub_allocator;
-      cub_allocator = new cub::CachingDeviceAllocator(2, 6, 31, (size_t) -1,
+      cub_allocator = new cub::CachingDeviceAllocator(2, 6, 22, (size_t) -1,
           false, debug_);
     } catch (...) {
     }