memory.hpp

/*
 * See LICENSE for more information about licensing
 *
 * Copyright 2023
 * Author: Luis G. Leon-Vega <luis.leon@ieee.org>
 *         Diego Arturo Avila Torres <diego.avila@uned.cr>
 *
 */
#pragma once
#include <cynq/enums.hpp>
#include <cynq/execution-graph.hpp>
#include <cynq/status.hpp>
#include <memory>
 
namespace cynq {
class IMemory {
 public:
  virtual ~IMemory() = default;
  enum Type {
    None = 0,
    XRT
  };
  virtual Status Sync(const SyncType type) = 0;
  virtual Status Sync(std::shared_ptr<IExecutionGraph> graph,
                      const SyncType type);
  virtual size_t Size() = 0;
  template <typename T>
  std::shared_ptr<T> HostAddress() {
    return std::reinterpret_pointer_cast<T>(this->GetHostAddress());
  }
  template <typename T>
  std::shared_ptr<T> DeviceAddress() {
    return std::reinterpret_pointer_cast<T>(this->GetDeviceAddress());
  }
 
  static std::shared_ptr<IMemory> Create(IMemory::Type impl,
                                         const std::size_t size,
                                         uint8_t* hostptr, uint8_t* devptr,
                                         void* moverptr);
 
 protected:
  virtual std::shared_ptr<uint8_t> GetHostAddress() = 0;
  virtual std::shared_ptr<uint8_t> GetDeviceAddress() = 0;
};
}  // namespace cynq