Helper class usable as a Thrust CUDA execution policy that uses RMM for temporary memory allocation on the specified stream. More...
#include <exec_policy.hpp>
Inheritance diagram for rmm::exec_policy:
Collaboration diagram for rmm::exec_policy:
Public Member Functions | |
| exec_policy (cuda_stream_view stream=cuda_stream_default, rmm::mr::device_memory_resource *mr=mr::get_current_device_resource()) | |
| Construct a new execution policy object. More... | |
Detailed Description
Helper class usable as a Thrust CUDA execution policy that uses RMM for temporary memory allocation on the specified stream.
Constructor & Destructor Documentation
◆ exec_policy()
|
inlineexplicit |
Construct a new execution policy object.
- Parameters
-
stream The stream on which to allocate temporary memory mr The resource to use for allocating temporary memory
The documentation for this class was generated from the following file:
