Use synchronize_optional for device-to-device copy in thrust::copy() (#3149)

davidwendt · web-flow · commit a9cefd9b0770 · 2024-12-13T14:02:54.000+01:00
diff --git 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;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -174,7 +174,7 @@ trivial_copy_device_to_device(Policy& policy, Type* dst, Type const* src, size_t`
`174`	`174`	`cudaStream_t stream = cuda_cub::stream(policy);`
`175`	`175`	`//`
`176`	`176`	`status = ::cudaMemcpyAsync(dst, src, sizeof(Type) * count, cudaMemcpyDeviceToDevice, stream);`
`177`		`- cuda_cub::synchronize(policy);`
	`177`	`+ cuda_cub::synchronize_optional(policy);`
`178`	`178`	`return status;`
`179`	`179`	`}`
`180`	`180`