VariableRef.hpp

// Copyright (c) 2018 Graphcore Ltd. All rights reserved.
 
#ifndef poplar_VariableRef_hpp
#define poplar_VariableRef_hpp
 
#include <poplar/Interval.hpp>
#include <poplar/Tensor.hpp>
 
namespace poplar {
 
class VariableRef {
  unsigned id;
 
public:
  VariableRef(unsigned id) : id(id) {}
  VariableRef() = default;
  VariableRef(const VariableRef &other) = default;
  VariableRef(VariableRef &&other) = default;
  VariableRef &operator=(const VariableRef &other) = default;
  VariableRef &operator=(VariableRef &&other) = default;
 
  friend bool operator==(const VariableRef &a, const VariableRef &b) {
    return a.id == b.id;
  }
 
  friend bool operator<(const VariableRef &a, const VariableRef &b) {
    return a.id < b.id;
  }
 
  std::size_t hash() const;
 
  friend class Graph;
};
 
struct VariableInterval {
  VariableRef var;
  Interval interval;
  VariableInterval(VariableRef var, Interval interval)
      : var(var), interval(std::move(interval)) {}
  VariableInterval() = default;
  VariableInterval(const VariableInterval &other) = default;
  VariableInterval(VariableInterval &&other) = default;
  VariableInterval &operator=(const VariableInterval &other) = default;
  VariableInterval &operator=(VariableInterval &&other) = default;
};
 
bool operator==(const VariableInterval &a, const VariableInterval &b);
bool operator<(const VariableInterval &a, const VariableInterval &b);
 
} // end namespace poplar
 
namespace std {
 
template <> struct hash<poplar::VariableRef> {
  size_t operator()(const poplar::VariableRef &v) const { return v.hash(); }
};
 
} // namespace std
 
#endif // poplar_VariableRef_hpp