#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>
类 | |
class | rmm::exec_policy |
可用作 Thrust CUDA 执行策略的辅助类,它使用 RMM 在指定的流上分配临时内存。 更多... | |
类型定义 | |
using | rmm::thrust_exec_policy_t = thrust::detail::execute_with_allocator< mr::thrust_allocator< char >, thrust::cuda_cub::execute_on_stream_base > |
使用 thrust 进行分配的同步执行策略。 | |
using | rmm::thrust_exec_policy_nosync_t = thrust_exec_policy_t |
using | rmm::exec_policy_nosync = exec_policy |
与 Thrust < 1.16.0 一起使用时,exec_policy_nosync 是 exec_policy 的别名。 | |
使用 RMM 的 Thrust Allocator Adaptor 的 Thrust 执行策略。