From: Jonathan L Long <jonlong@cs.berkeley.edu>
Date: Wed, 17 Dec 2014 04:18:53 +0000 (-0800)
Subject: check host malloc result
X-Git-Tag: submit/tizen/20180823.020014~572^2~80^2
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=da18d318f4d99036a61fa6c3d843eb43827268b8;p=platform%2Fupstream%2Fcaffeonacl.git

check host malloc result
---

diff --git a/include/caffe/syncedmem.hpp b/include/caffe/syncedmem.hpp
index db8d0e8..2564e07 100644
--- a/include/caffe/syncedmem.hpp
+++ b/include/caffe/syncedmem.hpp
@@ -24,6 +24,7 @@ namespace caffe {
 
 inline void CaffeMallocHost(void** ptr, size_t size) {
   *ptr = malloc(size);
+  CHECK(*ptr) << "host allocation of size " << size << " failed";
 }
 
 inline void CaffeFreeHost(void* ptr) {