allocator.h

Go to the documentation of this file.

#ifndef BASE_CORE_ALLOCATOR_H_
#define BASE_CORE_ALLOCATOR_H_
 
#include <ATen/core/tensor_types.h>
 
namespace base {
namespace core {
class Allocator {
  public:
    virtual void* allocate(size_t size) = 0;
 
    virtual void* allocate(size_t size, size_t alignment) = 0;
 
    virtual void free(void* ptr) = 0;
 
    virtual size_t AllocatedSize(void* ptr) {
        return allocated_size_;
    }
 
    virtual container::DeviceType GetDeviceType() = 0;
 
    virtual ~Allocator() = default;
 
  protected:
    size_t allocated_size_ = 0;
};
 
} // namespace core
} // namespace base
 
#endif // BASE_CORE_ALLOCATOR_H_