mlir_gen.h

Go to the documentation of this file.

#pragma once
 
#include <mlir/IR/MLIRContext.h>
#include <unordered_map>
#include <string>
#include <vector>
 
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/IR/Dialect.h"
#include "mlir/Pass/PassManager.h"
 
#include "dialect.h"
#include "types.h"
#include "parser.h"
 
namespace picceler {

class MLIRGen {
public:
  using GeneratorFunction = std::function<mlir::Value(mlir::Location, const std::vector<mlir::Value> &)>;
  using VariableTable = std::unordered_map<std::string, mlir::Value>;
  using NamedVariableTable = std::pair<std::string, VariableTable>;
 
public:
  MLIRGen() = delete;
  MLIRGen(mlir::MLIRContext *context);
  MLIRGen(const MLIRGen &) = default;
  MLIRGen &operator=(const MLIRGen &) = default;
  MLIRGen(MLIRGen &&) = default;
  MLIRGen &operator=(MLIRGen &&) = default;
  ~MLIRGen() noexcept;

  mlir::ModuleOp generate(ModuleNode *root, const std::string &sourceFile);
 
private:
  void declareUserFunctions(mlir::ModuleOp module, ModuleNode *root);

  void defineUserFunctions(mlir::ModuleOp module, ModuleNode *root);

  std::vector<mlir::Type> getFunctionArgTypes(FunctionNode *funcNode);

  mlir::Type getMLIRType(const std::string &typeName);

  Result<mlir::Value> lookupVariable(const std::string &name) const;

  void declareVariable(const std::string &name, mlir::Value value);

  void enterScope(const std::string &name);

  void exitScope() noexcept;

  void emitStatement(ASTNode *node);

  mlir::Value emitKernel(KernelNode *node);
  mlir::Value emitExpression(ASTNode *node);
  mlir::Value emitAssignment(AssignmentNode *node);
  mlir::Value emitCall(CallNode *node);
  mlir::Value emitVariable(VariableNode *node);
  mlir::Value emitString(StringNode *node);
  mlir::Value emitNumber(NumberNode *node);
  mlir::Value emitBinaryOp(BinaryOpNode *node);
  void emitIf(IfNode *node);
  void emitFor(ForNode *node);
  void emitReturn(ReturnNode *node);
  mlir::Value emitCallExpression(CallNode *node, const std::vector<mlir::Value> &args);


  void registerBuiltinFunctions();
 
private:
  mlir::MLIRContext *_context;
  mlir::OpBuilder _builder;
  std::vector<NamedVariableTable> _scopedVariableTable;
  std::unordered_map<std::string, GeneratorFunction> _functionTable;
  std::string _sourceFile;
};
 
} // namespace picceler