diff options
author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-06-27 18:36:48 -0700 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-07-03 17:14:11 -0700 |
commit | 9f74b6b129ab6ae161d569ce73d346553e04ec49 (patch) | |
tree | 0183412df4de67935a86c90b92f5639f626fd6de /src/caffe/layers/hdf5_output_layer.cu | |
parent | 1fcd29fb9aaedb763f58a1702dedc4a1e3904422 (diff) | |
download | caffeonacl-9f74b6b129ab6ae161d569ce73d346553e04ec49.tar.gz caffeonacl-9f74b6b129ab6ae161d569ce73d346553e04ec49.tar.bz2 caffeonacl-9f74b6b129ab6ae161d569ce73d346553e04ec49.zip |
switch to unified virtual addressing CUDA memcpy
Host / device copies are distinguished by the virtual address of the
pointers instead of explicit memcpy modes.
Diffstat (limited to 'src/caffe/layers/hdf5_output_layer.cu')
-rw-r--r-- | src/caffe/layers/hdf5_output_layer.cu | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/caffe/layers/hdf5_output_layer.cu b/src/caffe/layers/hdf5_output_layer.cu index 59505ee6..19567c12 100644 --- a/src/caffe/layers/hdf5_output_layer.cu +++ b/src/caffe/layers/hdf5_output_layer.cu @@ -29,10 +29,10 @@ Dtype HDF5OutputLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, for (int i = 0; i < bottom[0]->num(); ++i) { CUDA_CHECK(cudaMemcpy(&data_blob_.mutable_cpu_data()[i * data_datum_dim], &bottom[0]->gpu_data()[i * data_datum_dim], - sizeof(Dtype) * data_datum_dim, cudaMemcpyDeviceToHost)); + sizeof(Dtype) * data_datum_dim, cudaMemcpyDefault)); CUDA_CHECK(cudaMemcpy(&label_blob_.mutable_cpu_data()[i * label_datum_dim], &bottom[1]->gpu_data()[i * label_datum_dim], - sizeof(Dtype) * label_datum_dim, cudaMemcpyDeviceToHost)); + sizeof(Dtype) * label_datum_dim, cudaMemcpyDefault)); } SaveBlobs(); return Dtype(0.); |