LLZKLoweringUtils.h

Go to the documentation of this file.

//===-- LLZKLoweringUtils.h -------------------------------------*- C++ -*-===//
//
// Shared utilities for lowering passes in the LLZK project.
//
//===----------------------------------------------------------------------===//
 
#ifndef LLZK_TRANSFORMS_LOWERING_UTILS_H
#define LLZK_TRANSFORMS_LOWERING_UTILS_H
 
#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/Value.h>
#include <mlir/Support/LogicalResult.h>
 
#include <llvm/ADT/DenseMap.h>
 
namespace llzk {
 
struct AuxAssignment {
  std::string auxFieldName;
  mlir::Value computedValue;
};
 
mlir::Value getSelfValueFromCompute(function::FuncDefOp computeFunc);
 
mlir::Value rebuildExprInCompute(
    mlir::Value val, function::FuncDefOp computeFunc, mlir::OpBuilder &builder,
    llvm::DenseMap<mlir::Value, mlir::Value> &memo
);
 
mlir::LogicalResult
checkForAuxFieldConflicts(component::StructDefOp structDef, llvm::StringRef auxPrefix);
 
component::FieldDefOp addAuxField(component::StructDefOp structDef, llvm::StringRef name);
 
unsigned getFeltDegree(mlir::Value val, llvm::DenseMap<mlir::Value, unsigned> &memo);

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