LLZKLoweringUtils.h

Go to the documentation of this file.

//===-- LLZKLoweringUtils.h -------------------------------------*- C++ -*-===//
//
// Shared utilities for lowering passes in the LLZK project.
//
//===----------------------------------------------------------------------===//
 
#pragma once
 
#include "llzk/Dialect/Constrain/IR/Ops.h"
#include "llzk/Dialect/Felt/IR/Ops.h"
#include "llzk/Dialect/Function/IR/Ops.h"
 
#include <mlir/IR/Builders.h>
#include <mlir/IR/BuiltinOps.h>
#include <mlir/IR/Types.h>
#include <mlir/IR/Value.h>
#include <mlir/Support/LogicalResult.h>
 
#include <llvm/ADT/DenseMap.h>
 
namespace llzk {
 
struct AuxAssignment {
  std::string auxMemberName;
  mlir::Value computedValue;
};

mlir::Value rebuildExprInCompute(
    mlir::Value val, function::FuncDefOp computeFunc, mlir::OpBuilder &builder,
    llvm::DenseMap<mlir::Value, mlir::Value> &memo
);
 
mlir::LogicalResult
checkForAuxMemberConflicts(component::StructDefOp structDef, llvm::StringRef auxPrefix);

mlir::LogicalResult checkFuncBodyIsStraightLine(function::FuncDefOp func, llvm::StringRef passName);
 
component::MemberDefOp
addAuxMember(component::StructDefOp structDef, llvm::StringRef name, mlir::Type type);
 
unsigned getFeltDegree(mlir::Value val, llvm::DenseMap<mlir::Value, unsigned> &memo);

void replaceSubsequentUsesWith(mlir::Value oldVal, mlir::Value newVal, mlir::Operation *afterOp);
 
} // namespace llzk