From 09868aced72a20bc30198b84a225e58fdd6f13a6 Mon Sep 17 00:00:00 2001
From: Ronghang Hu <huronghang@hotmail.com>
Date: Sat, 15 Aug 2015 19:02:57 -0700
Subject: [PATCH] Malloc at least 1 byte for MultiGPU P2PSync buffers

---
 src/caffe/parallel.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/caffe/parallel.cpp b/src/caffe/parallel.cpp
index d48136c56b8..a6d154e168e 100644
--- 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