|
void * | allocate (size_type num_bytes) override |
|
void | deallocate (void *ptr) override |
|
| CudaAsyncAllocator (CUstream_st *stream) |
|
bool | check_environment (int device_id, CUstream_st *stream) const override |
| Checks if the allocator can be used safely with the provided device ID and stream.
|
|
◆ allocate()
void * gko::CudaAsyncAllocator::allocate |
( |
size_type | num_bytes | ) |
|
|
overridevirtual |
◆ check_environment()
bool gko::CudaAsyncAllocator::check_environment |
( |
int | device_id, |
|
|
CUstream_st * | stream ) const |
|
overridevirtual |
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::CudaAsyncAllocator::deallocate |
( |
void * | ptr | ) |
|
|
overridevirtual |
The documentation for this class was generated from the following file: