From f19deb49677851232eb4409de067fe34eef2e460 Mon Sep 17 00:00:00 2001 From: David Wendt Date: Thu, 12 Dec 2024 17:42:57 -0500 Subject: [PATCH] Use synchronize_optional for device-to-device copy in thrust::copy() --- thrust/thrust/system/cuda/detail/util.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/thrust/thrust/system/cuda/detail/util.h b/thrust/thrust/system/cuda/detail/util.h index 7185ddb8d7e..b5a7125e61b 100644 --- a/thrust/thrust/system/cuda/detail/util.h +++ b/thrust/thrust/system/cuda/detail/util.h @@ -174,7 +174,7 @@ trivial_copy_device_to_device(Policy& policy, Type* dst, Type const* src, size_t cudaStream_t stream = cuda_cub::stream(policy); // status = ::cudaMemcpyAsync(dst, src, sizeof(Type) * count, cudaMemcpyDeviceToDevice, stream); - cuda_cub::synchronize(policy); + cuda_cub::synchronize_optional(policy); return status; }