From 43a2595f5d3b8239600f205206d390ec62a4a2d9 Mon Sep 17 00:00:00 2001
From: Feng Tian <feng.tian@intel.com>
Date: Fri, 9 Mar 2018 16:08:38 +0800
Subject: [PATCH] revert changes made in c4a0ad1f3 as it causes iteration time
 increase

---
 src/caffe/syncedmem.cpp | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

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<PrvMemDescr> 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;