LLZKComputeConstrainToProductPass.h

Go to the documentation of this file.

//===-- LLZKComputeConstrainToProductPass.h ---------------------*- C++ -*-===//
//
// Part of the LLZK Project, under the Apache License v2.0.
// See LICENSE.txt for license information.
// Copyright 2025 Veridise Inc.
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
 
#pragma once
 
#include "llzk/Analysis/LightweightSignalEquivalenceAnalysis.h"
#include "llzk/Dialect/Function/IR/Ops.h"
#include "llzk/Dialect/Struct/IR/Ops.h"
 
#include <mlir/Support/LogicalResult.h>
 
#include <vector>
 
namespace llzk {
 
class ProductAligner {
  mlir::SymbolTableCollection &tables;
  LightweightSignalEquivalenceAnalysis &equivalence;
 
public:
  std::vector<component::StructDefOp> alignedStructs;
  ProductAligner(
      mlir::SymbolTableCollection &_tables, LightweightSignalEquivalenceAnalysis &_equivalence
  )
      : tables {_tables}, equivalence {_equivalence} {}
 
  // Given a @product function body, try to match up calls to @A::@compute and @A::@constrain for
  // every sub-struct @A and replace them with a call to @A::@product
  mlir::LogicalResult alignCalls(function::FuncDefOp product);
 
  // Given a StructDefOp @root, replace the @root::@compute and @root::@constrain functions with a
  // @root::@product
  function::FuncDefOp alignFuncs(
      component::StructDefOp root, function::FuncDefOp compute, function::FuncDefOp constrain
  );
};
 
} // namespace llzk