From 30178485540bea28d58c276babbad63f5353792e Mon Sep 17 00:00:00 2001 From: Yangqing Jia Date: Mon, 28 Oct 2013 11:29:03 -0700 Subject: [PATCH] syncedmem bugfix --- include/caffe/syncedmem.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/caffe/syncedmem.hpp b/include/caffe/syncedmem.hpp index 1229adb..539f0d4 100644 --- a/include/caffe/syncedmem.hpp +++ b/include/caffe/syncedmem.hpp @@ -20,9 +20,9 @@ inline void CaffeMallocHost(void** ptr, size_t size) { inline void CaffeFreeHost(void* ptr) { if (Caffe::mode() == Caffe::GPU) { - CUDA_CHECK(cudaFreeHost(ptr))); + CUDA_CHECK(cudaFreeHost(ptr)); } else { - free(ptr) + free(ptr); } } -- 2.7.4