All Classes Files Functions Variables Typedefs Enumerations Enumerator Friends Modules Pages
Classes | Functions
polymorphic_allocator.hpp File Reference
#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>
Include dependency graph for polymorphic_allocator.hpp:

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_allocators 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_allocators 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_adaptors 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_adaptors 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...