CaffeMallocHost(&cpu_ptr_, size_);
own_cpu_data_ = true;
}
- Caffe::set_mode(Caffe::GPU);
- caffe_memcpy(size_, gpu_ptr_, cpu_ptr_);
+ CUDA_CHECK(cudaMemcpy(cpu_ptr_, gpu_ptr_, size_, cudaMemcpyDefault));
head_ = SYNCED;
break;
case HEAD_AT_CPU:
if (gpu_ptr_ == NULL) {
CUDA_CHECK(cudaMalloc(&gpu_ptr_, size_));
}
- Caffe::set_mode(Caffe::GPU);
- caffe_memcpy(size_, cpu_ptr_, gpu_ptr_);
+ CUDA_CHECK(cudaMemcpy(gpu_ptr_, cpu_ptr_, size_, cudaMemcpyDefault));
head_ = SYNCED;
break;
case HEAD_AT_GPU: