wip

[jacinto-ai/caffe-jacinto.git] / src / caffe / blob.cpp

diff --git a/src/caffe/blob.cpp b/src/caffe/blob.cpp
index 4d610aef4f6eadd6931d67542200ca625a75fa2a..6695333b429e69722ff33e39a42d39648757d83a 100644 (file)
--- a/src/caffe/blob.cpp
+++ b/src/caffe/blob.cpp
@@ -246,7 +246,7 @@ void Blob::CopyFrom(const Blob& source, bool copy_diff, bool reshape,
     CUDNN_CHECK(cudnnTransformTensor(handle,
         cudnn::one(src_type), src_desc, src->gpu_data(),
         cudnn::zero(dst_type), dst_desc, dst->mutable_gpu_data(false)));
-    CUDA_CHECK(cudaStreamSynchronize(Caffe::thread_stream()));
+    CUDA_CHECK(cudaStreamSynchronize(Caffe::thread_stream(group)));
     CUDNN_CHECK(cudnnDestroyTensorDescriptor(src_desc));
     CUDNN_CHECK(cudnnDestroyTensorDescriptor(dst_desc));
   }