From 1d820f7fa81e6b02a64140a93edbfd30dc529e8b Mon Sep 17 00:00:00 2001
From: Ronghang Hu <huronghang@hotmail.com>
Date: Fri, 14 Aug 2015 09:19:48 -0700
Subject: [PATCH] Malloc at least one byte in Parallel

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

diff --git a/src/caffe/parallel.cpp b/src/caffe/parallel.cpp
index 6e7d802bb99..d48136c56b8 100644
--- a/src/caffe/parallel.cpp
+++ b/src/caffe/parallel.cpp
@@ -64,7 +64,9 @@ static size_t total_size(const vector<Blob<Dtype>*>& params) {
   size_t size = 0;
   for (int i = 0; i < params.size(); ++i)
     size += params[i]->count();
-  return size;
+  // Size have at least one byte, otherwise cudaMalloc fails if net has no
+  // learnable parameters.
+  return (size > 0) ? size : 1;
 }
 
 template<typename Dtype>