Skip to content
This repository was archived by the owner on Aug 5, 2022. It is now read-only.

Commit 43a2595

Browse files
committed
revert changes made in c4a0ad1 as it causes iteration time increase
1 parent 2a30fe7 commit 43a2595

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/caffe/syncedmem.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -263,10 +263,8 @@ void SyncedMemory::set_prv_descriptor(shared_ptr<PrvMemDescr> descriptor,
263263
} else {
264264
if ((head_ != HEAD_AT_PRV) && same_data)
265265
head_ = SYNCED_PRV;
266-
else if (descriptor->prv_ptr())
267-
head_ = HEAD_AT_PRV;
268266
else
269-
head_ = HEAD_AT_CPU;
267+
head_ = HEAD_AT_PRV;
270268
}
271269

272270
prv_descriptor_ = descriptor;

0 commit comments

Comments
 (0)