diff options
author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-06-27 19:51:33 -0700 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-07-03 17:14:12 -0700 |
commit | 1da30a555827e778e36c5db72ef9cc672ad85675 (patch) | |
tree | aeee78e1972481269f16eba025c3b053a8d13cea /src/caffe/layers/hdf5_data_layer.cu | |
parent | aaa9f24d369798cfa268f465ec5a1c35257ed500 (diff) | |
download | caffeonacl-1da30a555827e778e36c5db72ef9cc672ad85675.tar.gz caffeonacl-1da30a555827e778e36c5db72ef9cc672ad85675.tar.bz2 caffeonacl-1da30a555827e778e36c5db72ef9cc672ad85675.zip |
replace all memcpy by caffe_copy
Diffstat (limited to 'src/caffe/layers/hdf5_data_layer.cu')
-rw-r--r-- | src/caffe/layers/hdf5_data_layer.cu | 16 |
1 files changed, 6 insertions, 10 deletions
diff --git a/src/caffe/layers/hdf5_data_layer.cu b/src/caffe/layers/hdf5_data_layer.cu index 232f55fd..3c27f37e 100644 --- a/src/caffe/layers/hdf5_data_layer.cu +++ b/src/caffe/layers/hdf5_data_layer.cu @@ -40,16 +40,12 @@ Dtype HDF5DataLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, } current_row_ = 0; } - CUDA_CHECK(cudaMemcpy( - &(*top)[0]->mutable_gpu_data()[i * data_count], - &data_blob_.cpu_data()[current_row_ * data_count], - sizeof(Dtype) * data_count, - cudaMemcpyDefault)); - CUDA_CHECK(cudaMemcpy( - &(*top)[1]->mutable_gpu_data()[i * label_data_count], - &label_blob_.cpu_data()[current_row_ * label_data_count], - sizeof(Dtype) * label_data_count, - cudaMemcpyDefault)); + caffe_copy(data_count, + &data_blob_.cpu_data()[current_row_ * data_count], + &(*top)[0]->mutable_gpu_data()[i * data_count]); + caffe_copy(label_data_count, + &label_blob_.cpu_data()[current_row_ * label_data_count], + &(*top)[1]->mutable_gpu_data()[i * label_data_count]); } return Dtype(0.); } |