abacus-develop/cpu__allocator_8h_source.html

#ifndef BASE_CORE_CPU_ALLOCATOR_H_

#define BASE_CORE_CPU_ALLOCATOR_H_


#include <base/core/allocator.h>


namespace base {

namespace core {


class CPUAllocator : 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 core

} // namespace base


#endif // BASE_CORE_CPU_ALLOCATOR_H_

allocator.h

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

base::core::CPUAllocator
An Allocator subclass for CPU memory.
Definition cpu_allocator.h:16

base::core::CPUAllocator::GetDeviceType
container::DeviceType GetDeviceType() override
Get the type of device used by the TensorBuffer.
Definition cpu_allocator.cpp:29

base::core::CPUAllocator::free
void free(void *ptr) override
Free a block of CPU memory that was previously allocated by this allocator.
Definition cpu_allocator.cpp:23

base::core::CPUAllocator::allocate
void * allocate(size_t size) override
Allocate a block of CPU memory with the given size and default alignment.
Definition cpu_allocator.cpp:7

base
Definition allocator.h:6

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