EngineOptions.hpp

// Copyright (c) 2022 Graphcore Ltd. All rights reserved.
 
#ifndef poplar_EngineOptions_hpp
#define poplar_EngineOptions_hpp
 
#include <poplar/OptionFlags.hpp>
#include <poplar/Target.hpp>
 
namespace poplar {
 
namespace core {
struct EngineOptions;
}
 
class EngineOptions {
public:
  // Construct a EngineOptions object from an option flags object.
  // Flags from the environment variable are parsed and take precedence.
  explicit EngineOptions(const poplar::OptionFlags &opts, const Target &target,
                         unsigned repFactor);
 
  ~EngineOptions();
 
  const core::EngineOptions *getImpl() const { return impl.get(); }
 
private:
  std::unique_ptr<core::EngineOptions> impl;
  size_t targetHash;
  friend struct std::hash<EngineOptions>;
};
 
std::size_t hash_value(const EngineOptions &options);
 
} // namespace poplar
 
// Hashing support for EngineOptions
namespace std {
 
template <> struct hash<poplar::EngineOptions> {
  size_t operator()(const poplar::EngineOptions &) const;
};
 
} // namespace std
 
#endif // poplar_EngineOptions_hpp