NodeReduction.hpp

Go to the documentation of this file.

/*
 * Copyright 2018 Nico Reißmann <nico.reissmann@gmail.com>
 * See COPYING for terms of redistribution.
 */
 
#ifndef JLM_LLVM_OPT_NODEREDUCTION_HPP
#define JLM_LLVM_OPT_NODEREDUCTION_HPP
 
#include <jlm/rvsdg/simple-node.hpp>
#include <jlm/rvsdg/Transformation.hpp>
#include <jlm/util/Statistics.hpp>
 
#include <optional>
 
namespace jlm::rvsdg
{
class GammaNode;
class Graph;
class Node;
class Region;
class Output;
class StructuralNode;
class ThetaNode;
}
 
namespace jlm::llvm
{
 
class NodeReduction final : public rvsdg::Transformation
{
public:
  class Statistics;
 
  ~NodeReduction() noexcept override;
 
  NodeReduction();
 
  NodeReduction(const NodeReduction &) = delete;
 
  NodeReduction(NodeReduction &&) = delete;
 
  NodeReduction &
  operator=(const NodeReduction &) = delete;
 
  NodeReduction &
  operator=(NodeReduction &&) = delete;
 
  void
  Run(rvsdg::RvsdgModule & rvsdgModule, util::StatisticsCollector & statisticsCollector) override;
 
private:
  void
  ReduceNodesInRegion(rvsdg::Region & region);
 
  bool
  ReduceStructuralNode(rvsdg::StructuralNode & structuralNode);
 
  bool
  ReduceGammaNode(rvsdg::GammaNode & gammaNode);
 
  bool
  reduceThetaNode(rvsdg::ThetaNode & thetaNode);
 
  bool
  ReduceSimpleNode(rvsdg::SimpleNode & simpleNode);
 
  std::unique_ptr<Statistics> Statistics_;
};
 
class NodeReduction::Statistics final : public util::Statistics
{
  static inline const char * NumRegionsLabel_ = "NumRegions";
  static inline const char * NumTotalRegionIterationsLabel_ = "NumTotalRegionIterations";
  static inline const char * MaxIterationsPerRegionLabel_ = "MaxIterationsPerRegion";
 
public:
  ~Statistics() noexcept override = default;
 
  explicit Statistics(const util::FilePath & sourceFile)
      : util::Statistics(Id::ReduceNodes, sourceFile)
  {}
 
  void
  Start(const rvsdg::Graph & graph) noexcept;
 
  void
  End(const rvsdg::Graph & graph) noexcept;
 
  bool
  AddIteration(const rvsdg::Region & region, size_t numIterations);
 
  std::optional<size_t>
  GetNumIterations(const rvsdg::Region & region) const noexcept;
 
  [[nodiscard]] size_t
  getNumRegions() const noexcept;
 
  [[nodiscard]] size_t
  getTotalIterations() const noexcept;
 
  [[nodiscard]] size_t
  getMaxIterationsPerRegion() const noexcept;
 
  struct ReductionCounters
  {
    size_t numLoadNonVolatileReductions = 0;
    size_t numStoreNonVolatileReductions = 0;
 
    size_t numMemoryStateMergeReductions = 0;
    size_t numMemoryStateJoinReductions = 0;
    size_t numMemoryStateSplitReductions = 0;
    size_t numLambdaExitMemoryStateMergeReductions = 0;
 
    size_t numMatchReductions = 0;
    size_t numSExtReductions = 0;
    size_t numZExtReductions = 0;
    size_t numTruncReductions = 0;
    size_t numFPExtReductions = 0;
    size_t numFPTruncReductions = 0;
 
    size_t numIntegerEqReductions = 0;
    size_t numIntegerNeReductions = 0;
    size_t numIntegerSgeReductions = 0;
    size_t numIntegerSgtReductions = 0;
    size_t numIntegerSleReductions = 0;
    size_t numIntegerSltReductions = 0;
    size_t numIntegerUgeReductions = 0;
    size_t numIntegerUgtReductions = 0;
    size_t numIntegerUleReductions = 0;
    size_t numIntegerUltReductions = 0;
 
    size_t numIntegerAddReductions = 0;
    size_t numIntegerSubReductions = 0;
    size_t numIntegerMulReductions = 0;
    size_t numIntegerSDivReductions = 0;
    size_t numIntegerUDivReductions = 0;
    size_t numIntegerSRemReductions = 0;
    size_t numIntegerURemReductions = 0;
    size_t numIntegerAShrReductions = 0;
    size_t numIntegerShlReductions = 0;
    size_t numIntegerLShrReductions = 0;
    size_t numIntegerAndReductions = 0;
    size_t numIntegerOrReductions = 0;
    size_t numIntegerXorReductions = 0;
 
    size_t numFPBinaryOpReductions = 0;
 
    size_t numPtrCmpReductions = 0;
    size_t numGetElementPtrReductions = 0;
    size_t numFCmpReductions = 0;
    size_t numMemoryHoistBarrierReductions = 0;
    size_t numBinaryReductions = 0;
 
    size_t numGammaReductions = 0;
    size_t numThetaReductions = 0;
  };
 
  [[nodiscard]] ReductionCounters &
  getReductionCounters() noexcept
  {
    return reductionCounters_;
  }
 
  [[nodiscard]] const ReductionCounters &
  getReductionCounters() const noexcept
  {
    return reductionCounters_;
  }
 
  static std::unique_ptr<Statistics>
  Create(const util::FilePath & sourceFile)
  {
    return std::make_unique<Statistics>(sourceFile);
  }
 
private:
  ReductionCounters reductionCounters_{};
  std::unordered_map<const rvsdg::Region *, size_t> NumIterations_;
};
 
}
 
#endif