diff --git a/src/caffe/syncedmem.cpp b/src/caffe/syncedmem.cpp index 7925c63e9..507ccf150 100644 --- a/src/caffe/syncedmem.cpp +++ b/src/caffe/syncedmem.cpp @@ -263,10 +263,8 @@ void SyncedMemory::set_prv_descriptor(shared_ptr descriptor, } else { if ((head_ != HEAD_AT_PRV) && same_data) head_ = SYNCED_PRV; - else if (descriptor->prv_ptr()) - head_ = HEAD_AT_PRV; else - head_ = HEAD_AT_CPU; + head_ = HEAD_AT_PRV; } prv_descriptor_ = descriptor;