CudaAllocator Class Reference

CudaAllocator Class Reference#

Reference API: gko::CudaAllocator Class Reference
Reference API
gko::CudaAllocator Class Reference

#include <ginkgo/core/base/memory.hpp>

Inheritance diagram for gko::CudaAllocator:
[legend]

Public Member Functions

void * allocate (size_type num_bytes) override
 
void deallocate (void *ptr) override
 

Additional Inherited Members

- Protected Member Functions inherited from gko::CudaAllocatorBase
virtual bool check_environment (int device_id, CUstream_st *stream) const
 

Detailed Description

Allocator using cudaMalloc.

Member Function Documentation

◆ allocate()

void * gko::CudaAllocator::allocate ( size_type  num_bytes)
overridevirtual

Implements gko::Allocator.

◆ deallocate()

void gko::CudaAllocator::deallocate ( void *  ptr)
overridevirtual

Implements gko::Allocator.


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