mlir_gen.h

Go to the documentation of this file.

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

class MLIRGen {
public:
  MLIRGen(mlir::MLIRContext *context);

  mlir::ModuleOp generate(ModuleNode *root);
 
private:
  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 emitBuiltinCall(CallNode *node, const std::vector<mlir::Value> &args);


  bool isBuiltinFunction(CallNode *node);

  std::unordered_map<std::string, mlir::Value> builtinVariables();

  std::unordered_map<std::string, mlir::func::FuncOp> builtinFunctions();
 
private:
  mlir::MLIRContext *_context;
  mlir::OpBuilder _builder;
  std::unordered_map<std::string, mlir::Value> _variableTable;
  std::unordered_map<std::string, mlir::func::FuncOp> _functionTable;
};
 
} // namespace picceler