From 00433d8082796bdaa2cafa067c158a5f31245985 Mon Sep 17 00:00:00 2001
From: Kai Li <kaili_kloud@163.com>
Date: Wed, 9 Jul 2014 10:12:07 +0800
Subject: [PATCH] Switch to GPU mode when pointer is move to or from GPU in
 SyncedMemory

---
 src/caffe/syncedmem.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/caffe/syncedmem.cpp b/src/caffe/syncedmem.cpp
index 9fe5528..5d6d058 100644
--- a/src/caffe/syncedmem.cpp
+++ b/src/caffe/syncedmem.cpp
@@ -33,6 +33,7 @@ inline void SyncedMemory::to_cpu() {
       CaffeMallocHost(&cpu_ptr_, size_);
       own_cpu_data_ = true;
     }
+    Caffe::set_mode(Caffe::GPU);
     caffe_memcpy(size_, gpu_ptr_, cpu_ptr_);
     head_ = SYNCED;
     break;
@@ -53,6 +54,7 @@ inline void SyncedMemory::to_gpu() {
     if (gpu_ptr_ == NULL) {
       CUDA_CHECK(cudaMalloc(&gpu_ptr_, size_));
     }
+    Caffe::set_mode(Caffe::GPU);
     caffe_memcpy(size_, cpu_ptr_, gpu_ptr_);
     head_ = SYNCED;
     break;
-- 
2.7.4