GlobalExchangeConstraints.hpp

// Copyright (c) 2018 Graphcore Ltd. All rights reserved.
 
#ifndef poplar_GlobalExchangeConstraints_hpp
#define poplar_GlobalExchangeConstraints_hpp
#include <poplar/ArrayRef.hpp>
 
#include <functional>
#include <vector>
 
namespace poplar {
 
struct GlobalExchangeFlow {
  unsigned src;
  unsigned dst;
  GlobalExchangeFlow(unsigned src, unsigned dst) : src(src), dst(dst) {}
 
  bool operator==(const GlobalExchangeFlow &other) const;
 
  bool operator<(const GlobalExchangeFlow &other) const;
};
 
struct GlobalExchangeConstraint {
  double bandwidth;
  std::vector<GlobalExchangeFlow> flows;
 
  GlobalExchangeConstraint(double bandwidth, ArrayRef<GlobalExchangeFlow> flows)
      : bandwidth(bandwidth), flows(flows.begin(), flows.end()) {}
 
  bool operator==(const GlobalExchangeConstraint &other) const;
 
  bool operator<(const GlobalExchangeConstraint &other) const;
};
 
} // end namespace poplar
 
namespace std {
 
template <> struct hash<poplar::GlobalExchangeFlow> {
  size_t operator()(const poplar::GlobalExchangeFlow &value) const;
};
 
template <> struct hash<poplar::GlobalExchangeConstraint> {
  size_t operator()(const poplar::GlobalExchangeConstraint &value) const;
};
 
} // namespace std
 
#endif // poplar_GlobalExchangeConstraints_hpp