Preallocations.hpp

// Copyright (c) 2022 Graphcore Ltd. All rights reserved.
 
#ifndef poplar_Preallocations_hpp
#define poplar_Preallocations_hpp
 
#include "poplar/StringRef.hpp"
#include <cstddef>
#include <memory>
 
namespace poplar {
 
class Device;
class Target;
 
namespace core {
class Preallocations;
}
 
class Preallocations {
  std::unique_ptr<core::Preallocations> impl;
 
public:
  Preallocations(const Target &target);
  Preallocations(const Device &device);
  Preallocations(const Preallocations &);
  Preallocations(Preallocations &&) noexcept;
  ~Preallocations();
  void addAllocation(unsigned tile, std::size_t address, std::size_t size);
  void addAllocation(unsigned tile, std::size_t address, std::size_t size,
                     StringRef name);
 
  // Implementation
  const core::Preallocations &getImpl() const { return *impl; }
};
 
} // End namespace poplar.
 
#endif // poplar_Preallocations_hpp