#include <rmm/cuda_stream_view.hpp>
#include <rmm/detail/export.hpp>
#include <rmm/detail/thrust_namespace.h>
#include <rmm/mr/device/thrust_allocator_adaptor.hpp>
#include <rmm/resource_ref.hpp>
#include <thrust/system/cuda/execution_policy.h>
#include <thrust/version.h>
Go to the source code of this file.
Classes | |
class | rmm::exec_policy |
Helper class usable as a Thrust CUDA execution policy that uses RMM for temporary memory allocation on the specified stream. More... | |
class | rmm::exec_policy_nosync |
Helper class usable as a Thrust CUDA execution policy that uses RMM for temporary memory allocation on the specified stream and which allows the Thrust backend to skip stream synchronizations that are not required for correctness. More... | |
Typedefs | |
using | rmm::thrust_exec_policy_t = thrust::detail::execute_with_allocator< mr::thrust_allocator< char >, thrust::cuda_cub::execute_on_stream_base > |
Synchronous execution policy for allocations using Thrust. | |
using | rmm::thrust_exec_policy_nosync_t = thrust::detail::execute_with_allocator< mr::thrust_allocator< char >, thrust::cuda_cub::execute_on_stream_nosync_base > |
Asynchronous execution policy for allocations using Thrust. | |
Thrust execution policy that uses RMM's Thrust Allocator Adaptor.