CSRFunctions.hpp

// Copyright (c) 2019 Graphcore Ltd. All rights reserved.
 
#ifndef poplar_CSRFunctions_hpp
#define poplar_CSRFunctions_hpp
 
#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>
#include <string>
 
namespace poplar {
 
struct FloatingPointBehaviour {
  bool inv = true;
  bool div0 = true;
  bool oflo = true;
  bool esr = true;
  bool nanoo = true;
  FloatingPointBehaviour(bool inv, bool div0, bool oflo, bool esr, bool nanoo)
      : inv(inv), div0(div0), oflo(oflo), esr(esr), nanoo(nanoo) {}
  FloatingPointBehaviour() = default;
  FloatingPointBehaviour operator!() const {
    return FloatingPointBehaviour(!inv, !div0, !oflo, !esr, !nanoo);
  }
};
 
void setFloatingPointBehaviour(poplar::Graph &graph,
                               poplar::program::Sequence &prog,
                               const FloatingPointBehaviour &behaviour,
                               const DebugContext &debugContext = {});
 
void setFloatingPointBehaviour(poplar::Graph &graph,
                               poplar::program::Sequence &prog,
                               const poplar::Tensor &behaviour,
                               const DebugContext &debugContext = {});
 
void setStochasticRounding(poplar::Graph &graph,
                           poplar::program::Sequence &prog, bool behaviour,
                           const DebugContext &debugContext = {});
 
poplar::Tensor getAndModifyFloatingPointBehaviour(
    poplar::Graph &graph, poplar::program::Sequence &prog,
    const FloatingPointBehaviour &clear, const FloatingPointBehaviour &set,
    const DebugContext &debugContext = {});
 
} // namespace poplar
#endif // poplar_CSRFunctions_hpp