abacus-develop/gpu__allocator_8h_source.html

#ifndef BASE_CORE_GPU_ALLOCATOR_H_

#define BASE_CORE_GPU_ALLOCATOR_H_


#include <base/core/allocator.h>


namespace base {

namespace core {


class GPUAllocator : public Allocator {

public:

    void *allocate(size_t size) override;


    void *allocate(size_t size, size_t alignment) override;


    void free(void *ptr) override;


    container::DeviceType GetDeviceType() override;

};


} // namespace base

} // namespace core


#endif // BASE_CORE_GPU_ALLOCATOR_H_

allocator.h

base::core::Allocator
An abstract base class for memory allocators.
Definition allocator.h:17

base::core::GPUAllocator
An allocator that allocates memory on a GPU device.
Definition gpu_allocator.h:14

base::core::GPUAllocator::allocate
void * allocate(size_t size) override
Allocate a block of memory with the given size and default alignment on GPU.
Definition gpu_allocator.cpp:20

base::core::GPUAllocator::free
void free(void *ptr) override
Free a block of GPU memory that was previously allocated by this allocator.
Definition gpu_allocator.cpp:42

base::core::GPUAllocator::GetDeviceType
container::DeviceType GetDeviceType() override
Get the type of memory used by the TensorBuffer.
Definition gpu_allocator.cpp:48

base
Definition allocator.h:6

container::DeviceType
DeviceType
The type of memory used by an allocator.
Definition tensor_types.h:73