25 #include <rmm/detail/export.hpp>
26 #include <rmm/detail/thrust_namespace.h>
30 #include <thrust/system/cuda/execution_policy.h>
31 #include <thrust/version.h>
33 namespace RMM_NAMESPACE {
44 thrust::detail::execute_with_allocator<mr::thrust_allocator<char>,
45 thrust::cuda_cub::execute_on_stream_base>;
62 thrust::cuda::par(mr::thrust_allocator<char>(stream, mr)).on(stream.value()))
67 #if THRUST_VERSION >= 101600
73 thrust::detail::execute_with_allocator<mr::thrust_allocator<char>,
74 thrust::cuda_cub::execute_on_stream_nosync_base>;
86 thrust::cuda::par_nosync(mr::thrust_allocator<char>(stream, mr)).on(stream.value()))
Strongly-typed non-owning wrapper for CUDA streams with default constructor.
Definition: cuda_stream_view.hpp:41
Helper class usable as a Thrust CUDA execution policy that uses RMM for temporary memory allocation o...
Definition: exec_policy.hpp:51
exec_policy(cuda_stream_view stream=cuda_stream_default, device_async_resource_ref mr=mr::get_current_device_resource_ref())
Construct a new execution policy object.
Definition: exec_policy.hpp:59
static constexpr cuda_stream_view cuda_stream_default
Static cuda_stream_view of the default stream (stream 0), for convenience.
Definition: cuda_stream_view.hpp:127
cuda::mr::async_resource_ref< cuda::mr::device_accessible > device_async_resource_ref
Alias for a cuda::mr::async_resource_ref with the property cuda::mr::device_accessible.
Definition: resource_ref.hpp:41
device_async_resource_ref get_current_device_resource_ref()
Get the device_async_resource_ref for the current device.
Definition: per_device_resource.hpp:411
thrust_exec_policy_t thrust_exec_policy_nosync_t
Definition: exec_policy.hpp:95
thrust::detail::execute_with_allocator< mr::thrust_allocator< char >, thrust::cuda_cub::execute_on_stream_base > thrust_exec_policy_t
Synchronous execution policy for allocations using thrust.
Definition: exec_policy.hpp:45
exec_policy exec_policy_nosync
When used with Thrust < 1.16.0, exec_policy_nosync is an alias for exec_policy.
Definition: exec_policy.hpp:97