JlmToMlirConverter.hpp

Go to the documentation of this file.

/*
 * Copyright 2023 Magnus Sjalander <work@sjalander.com>
 * See COPYING for terms of redistribution.
 */
 
#ifndef JLM_MLIR_BACKEND_JLMTOMLIRCONVERTER_HPP
#define JLM_MLIR_BACKEND_JLMTOMLIRCONVERTER_HPP
 
// JLM
#include <jlm/llvm/ir/operators/delta.hpp>
#include <jlm/llvm/ir/operators/IntegerOperations.hpp>
#include <jlm/llvm/ir/operators/lambda.hpp>
#include <jlm/llvm/ir/operators/operators.hpp>
#include <jlm/llvm/ir/RvsdgModule.hpp>
#include <jlm/rvsdg/bitstring/arithmetic.hpp>
#include <jlm/rvsdg/gamma.hpp>
#include <jlm/rvsdg/theta.hpp>
 
// MLIR RVSDG dialects
#include <RVSDG/RVSDGDialect.h>
#include <RVSDG/RVSDGPasses.h>
 
// MLIR JLM dialects
#include <JLM/JLMDialect.h>
#include <JLM/JLMOps.h>
 
// MLIR generic dialects
#include <mlir/Dialect/Arith/IR/Arith.h>
#include <mlir/Dialect/LLVMIR/LLVMDialect.h>
 
namespace jlm::mlir
{
 
class JlmToMlirConverter final
{
public:
  JlmToMlirConverter()
      : Context_(std::make_unique<::mlir::MLIRContext>())
  {
    Context_->getOrLoadDialect<::mlir::rvsdg::RVSDGDialect>();
    Context_->getOrLoadDialect<::mlir::jlm::JLMDialect>();
    Context_->getOrLoadDialect<::mlir::arith::ArithDialect>();
    Context_->getOrLoadDialect<::mlir::LLVM::LLVMDialect>();
    Builder_ = std::make_unique<::mlir::OpBuilder>(Context_.get());
  }
 
  JlmToMlirConverter(const JlmToMlirConverter &) = delete;
 
  JlmToMlirConverter(JlmToMlirConverter &&) = delete;
 
  JlmToMlirConverter &
  operator=(const JlmToMlirConverter &) = delete;
 
  JlmToMlirConverter &
  operator=(JlmToMlirConverter &&) = delete;
 
  static void
  Print(::mlir::rvsdg::OmegaNode & omega, const util::FilePath & filePath);
 
  ::mlir::rvsdg::OmegaNode
  ConvertModule(const llvm::LlvmRvsdgModule & rvsdgModule);
 
  ::llvm::SmallVector<::mlir::Value>
  ConvertRegion(rvsdg::Region & region, ::mlir::Block & block, bool isRoot = false);
 
  static ::llvm::SmallVector<::mlir::Value>
  GetConvertedInputs(
      const rvsdg::Node & node,
      const std::unordered_map<rvsdg::Output *, ::mlir::Value> & valueMap);
 
  ::mlir::Operation *
  ConvertNode(
      const rvsdg::Node & node,
      ::mlir::Block & block,
      const ::llvm::SmallVector<::mlir::Value> & inputs);
 
  ::mlir::Operation *
  ConvertFpBinaryNode(
      const jlm::llvm::FBinaryOperation & op,
      ::llvm::SmallVector<::mlir::Value> inputs);
 
  ::mlir::Operation *
  ConvertFpCompareNode(
      const jlm::llvm::FCmpOperation & op,
      ::llvm::SmallVector<::mlir::Value> inputs);
 
  ::mlir::Operation *
  ConvertBitBinaryNode(
      const rvsdg::SimpleOperation & bitOp,
      ::llvm::SmallVector<::mlir::Value> inputs);
 
  ::mlir::Operation *
  ConvertIntegerBinaryOperation(
      const jlm::llvm::IntegerBinaryOperation & operation,
      ::llvm::SmallVector<::mlir::Value> inputs);
 
  ::mlir::Operation *
  BitCompareNode(const rvsdg::SimpleOperation & bitOp, ::llvm::SmallVector<::mlir::Value> inputs);
 
  ::mlir::Operation *
  ConvertPointerCompareNode(
      const llvm::PtrCmpOperation & pointerCompareOp,
      ::llvm::SmallVector<::mlir::Value> inputs);
 
  ::mlir::Operation *
  ConvertSimpleNode(
      const rvsdg::SimpleNode & node,
      ::mlir::Block & block,
      const ::llvm::SmallVector<::mlir::Value> & inputs);
 
  ::mlir::Operation *
  ConvertLambda(
      const rvsdg::LambdaNode & node,
      ::mlir::Block & block,
      const ::llvm::SmallVector<::mlir::Value> & inputs);
 
  ::mlir::Operation *
  ConvertGamma(
      const rvsdg::GammaNode & gammaNode,
      ::mlir::Block & block,
      const ::llvm::SmallVector<::mlir::Value> & inputs);
 
  ::mlir::Operation *
  ConvertTheta(
      const rvsdg::ThetaNode & thetaNode,
      ::mlir::Block & block,
      const ::llvm::SmallVector<::mlir::Value> & inputs);
 
  ::mlir::Operation *
  ConvertDelta(
      const rvsdg::DeltaNode & node,
      ::mlir::Block & block,
      const ::llvm::SmallVector<::mlir::Value> & inputs);
 
  ::mlir::FloatType
  ConvertFPType(const llvm::fpsize size);
 
  ::mlir::FunctionType
  ConvertFunctionType(const jlm::rvsdg::FunctionType & functionType);
 
  ::mlir::Type
  ConvertType(const rvsdg::Type & type);
 
  ::llvm::SmallVector<::mlir::Type>
  GetMemStateRange(size_t nresults);
 
  std::unique_ptr<::mlir::OpBuilder> Builder_;
  std::unique_ptr<::mlir::MLIRContext> Context_;
};
 
} // namespace jlm::mlir
 
#endif // JLM_MLIR_BACKEND_JLMTOMLIRCONVERTER_HPP