Malloc at least 1 byte for MultiGPU P2PSync buffers

sidestepism · Aug 16, 2015 · 09868ac · 09868ac
1 parent 25b4920
commit 09868ac
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/caffe/parallel.cpp b/src/caffe/parallel.cpp
@@ -55,7 +55,8 @@ static void apply_buffers(const vector<Blob<Dtype>*>& blobs,
     }
     ptr += size;
   }
-  CHECK_EQ(total_size, ptr - buffer);
+  // total_size is at least one byte
+  CHECK_EQ(total_size, (ptr == buffer ? 1 : ptr - buffer));
 }
 
 // Buffer size necessary to store given blobs