Malloc at least 1 byte for MultiGPU P2PSync buffers

diff --git a/src/caffe/parallel.cpp b/src/caffe/parallel.cpp
index d48136c..a6d154e 100644 (file)
--- 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