#include <rmm/cuda_stream_view.hpp>
#include <rmm/detail/export.hpp>
#include <rmm/mr/device/per_device_resource.hpp>
#include <rmm/resource_ref.hpp>
#include <cstddef>
#include <memory>
#include <type_traits>
Go to the source code of this file.
Classes | |
class | rmm::mr::polymorphic_allocator< T > |
A stream ordered Allocator using a rmm::mr::device_memory_resource to satisfy (de)allocations. More... | |
class | rmm::mr::stream_allocator_adaptor< Allocator > |
Adapts a stream ordered allocator to provide a standard Allocator interface. More... | |
struct | rmm::mr::stream_allocator_adaptor< Allocator >::rebind< T > |
Rebinds the allocator to the specified type. More... | |
Functions | |
template<typename T , typename U > | |
bool | rmm::mr::operator== (polymorphic_allocator< T > const &lhs, polymorphic_allocator< U > const &rhs) |
Compare two polymorphic_allocator s for equality. More... | |
template<typename T , typename U > | |
bool | rmm::mr::operator!= (polymorphic_allocator< T > const &lhs, polymorphic_allocator< U > const &rhs) |
Compare two polymorphic_allocator s for inequality. More... | |
template<typename A , typename O > | |
bool | rmm::mr::operator== (stream_allocator_adaptor< A > const &lhs, stream_allocator_adaptor< O > const &rhs) |
Compare two stream_allocator_adaptor s for equality. More... | |
template<typename A , typename O > | |
bool | rmm::mr::operator!= (stream_allocator_adaptor< A > const &lhs, stream_allocator_adaptor< O > const &rhs) |
Compare two stream_allocator_adaptor s for inequality. More... | |
template<typename Allocator > | |
auto | rmm::mr::make_stream_allocator_adaptor (Allocator const &allocator, cuda_stream_view stream) |
Factory to construct a stream_allocator_adaptor from an existing stream-ordered allocator. More... | |