Merge pull request #3788 from QiJune/fix_tensor_copy_bug

fix tensor copyfrom bug
revert-3824-remove_grad_op_type
QI JUN 8 years ago committed by GitHub
commit c2edd2dc07

@ -117,6 +117,8 @@ inline void Tensor::CopyFrom(const Tensor& src,
memory::Copy(boost::get<platform::GPUPlace>(dst_place), dst_ptr,
boost::get<platform::GPUPlace>(src_place), src_ptr, size, 0);
}
PADDLE_ENFORCE(cudaStreamSynchronize(0),
"cudaStreamSynchronize failed in Tensor CopyFrom");
#endif
}

Loading…
Cancel
Save