Add flag on how host memory is allocated
authorRonghang Hu <huronghang@hotmail.com>
Wed, 16 Sep 2015 20:39:23 +0000 (13:39 -0700)
committerRonghang Hu <huronghang@hotmail.com>
Fri, 25 Sep 2015 22:52:46 +0000 (15:52 -0700)
commitbd5f15427cc2f008f80378a5948ce379d93ebde6
tree419b87840e2824258e1d5df598dcac211ae4935f
parent71e05876f644a08af4cb1c955d01c5a272539e96
Add flag on how host memory is allocated

Add a bool flag to record whether a host memory is allocated using malloc or
cudaMallocHost, and free correspondingly using this flag, instead of depending on Caffe::mode(), which is mutable during runtime.
include/caffe/syncedmem.hpp
src/caffe/syncedmem.cpp