gko::CudaUnifiedAllocator Class Reference

Inheritance diagram for gko::CudaUnifiedAllocator:

Public Member Functions
void *	allocate (size_type num_bytes) override
void	deallocate (void *ptr) override
	CudaUnifiedAllocator (int device_id)
	CudaUnifiedAllocator (int device_id, unsigned int flags)

Protected Member Functions
bool	check_environment (int device_id, CUstream_st *stream) const override

Member Function Documentation

allocate()

void * gko::CudaUnifiedAllocator::allocate ( size_type  num_bytes )

overridevirtual

Implements gko::Allocator.

check_environment()

bool gko::CudaUnifiedAllocator::check_environment ( int  device_id, CUstream_st *  stream  ) const

overrideprotectedvirtual

Checks if the allocator can be used safely with the provided device ID and stream. The check is necessary to ensure safe usage of stream-ordered allocators and unified shared memory allocators.

Parameters

device_id	the device ID the allocator will be used in.
stream	the stream the allocator will be used with.

Returns

true if and only if the allocator can be used by CudaExecutor in the given environment.

Reimplemented from gko::CudaAllocatorBase.

deallocate()

void gko::CudaUnifiedAllocator::deallocate ( void *  ptr )

overridevirtual

Implements gko::Allocator.

---

The documentation for this class was generated from the following file:

• ginkgo/core/base/memory.hpp