From da18d318f4d99036a61fa6c3d843eb43827268b8 Mon Sep 17 00:00:00 2001
From: Jonathan L Long <jonlong@cs.berkeley.edu>
Date: Tue, 16 Dec 2014 20:18:53 -0800
Subject: [PATCH] check host malloc result

---
 include/caffe/syncedmem.hpp | 1 +
 1 file changed, 1 insertion(+)

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) {
-- 
2.7.4