PerfEstimateFunc.hpp

// Copyright (c) 2021 Graphcore Ltd. All rights reserved.
 
#ifndef poplar_PerfEstimateFunc_hpp
#define poplar_PerfEstimateFunc_hpp
#include <cstdint>
#include <functional>
 
namespace poplar {
class VertexIntrospector;
class Target;
 
struct VertexPerfEstimate {
  std::uint64_t cycles;
  std::uint64_t flops;
  VertexPerfEstimate() = default;
  VertexPerfEstimate(std::uint64_t cycles, std::uint64_t flops = 0)
      : cycles(cycles), flops(flops) {}
};
 
using PerfEstimateFunc = std::function<VertexPerfEstimate(
    const VertexIntrospector &v, const Target &target)>;
 
} // namespace poplar
 
#endif // poplar_PerfEstimateFunc_hpp