CudaAllocator Class Reference#
|
Reference API
|
gko::CudaAllocator Class Reference
#include <ginkgo/core/base/memory.hpp>
Inheritance diagram for gko::CudaAllocator:
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()
|
overridevirtual |
Implements gko::Allocator.
◆ deallocate()
|
overridevirtual |
Implements gko::Allocator.
The documentation for this class was generated from the following file:
- ginkgo/core/base/memory.hpp
Generated by
Protected Member Functions inherited from