Skip to content

Commit 875c6a1

Browse files
author
AdrianoDee
committed
Proper memcopy for localCoordToHostAsync
1 parent 4b75725 commit 875c6a1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

CUDADataFormats/TrackingRecHit/interface/TrackingRecHitSoADevice.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,10 @@ class TrackingRecHitSoADevice : public cms::cuda::PortableDeviceCollection<Track
4848
cms::cuda::host::unique_ptr<float[]> localCoordToHostAsync(cudaStream_t stream) const {
4949
auto ret = cms::cuda::make_host_unique<float[]>(4 * nHits(), stream);
5050
size_t rowSize = sizeof(float) * nHits();
51-
cudaCheck(cudaMemcpyAsync(ret.get(), view().xLocal(), rowSize * 4, cudaMemcpyDefault, stream));
5251

52+
size_t srcPitch = ptrdiff_t(view().yLocal()) - ptrdiff_t(view().xLocal());
53+
cudaCheck(cudaMemcpy2DAsync(ret.get(), rowSize, view().xLocal(), srcPitch, rowSize, 4, cudaMemcpyDeviceToHost, stream));
54+
5355
return ret;
5456
} //move to utilities
5557

0 commit comments

Comments
 (0)