TemplateInstantiation.h

Go to the documentation of this file.

//===-- TemplateInstantiation.h ---------------------------------*- C++ -*-===//
//
// Part of the LLZK Project, under the Apache License v2.0.
// See LICENSE.txt for license information.
// Copyright 2026 Veridise Inc.
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
 
#pragma once
 
#include "llzk/Dialect/Function/IR/Ops.h"
#include "llzk/Dialect/Polymorphic/IR/Ops.h"
#include "llzk/Dialect/Struct/IR/Ops.h"
#include "llzk/Util/TypeHelper.h"
 
#include <mlir/IR/Diagnostics.h>
#include <mlir/Transforms/DialectConversion.h>
 
#include <llvm/ADT/DenseMap.h>
#include <llvm/ADT/SmallVector.h>
 
#include <optional>
 
namespace llzk::polymorphic::detail {

class TemplateTypeConverter : public mlir::TypeConverter {
  llvm::DenseMap<mlir::Attribute, mlir::Attribute> paramNameToValue;
  mlir::Attribute convertIfPossible(mlir::Attribute attr) const;
 
public:
  explicit TemplateTypeConverter(llvm::DenseMap<mlir::Attribute, mlir::Attribute> bindings);

  mlir::Attribute convertAttr(mlir::Attribute attr) const;

  bool containsParam(mlir::Attribute name) const { return paramNameToValue.contains(name); }

  const llvm::DenseMap<mlir::Attribute, mlir::Attribute> &getParamMap() const {
    return paramNameToValue;
  }
};

mlir::FailureOr<mlir::ArrayAttr> rebaseTemplateParams(
    mlir::SymbolTableCollection &tables, mlir::ArrayAttr templateParams,
    mlir::Operation *lookupFrom, mlir::ModuleOp destinationRoot, mlir::Operation *requestSite
);

void reportDelayedDiagnostics(
    mlir::Operation *site, llvm::SmallVector<mlir::Diagnostic> &&diagnostics
);

void convertCalleesInPlace(
    mlir::Operation *op, const llvm::DenseMap<mlir::Attribute, mlir::Attribute> &bindings
);

void evaluateTemplateExprs(
    TemplateOp templateOp, llvm::DenseMap<mlir::Attribute, mlir::Attribute> &bindings
);

std::optional<mlir::Attribute>
inferUnifiedParam(const UnificationMap &unifyResult, mlir::SymbolRefAttr paramName);

mlir::LogicalResult substituteStructBody(
    component::StructDefOp clone, component::StructType originalType,
    const llvm::DenseMap<mlir::Attribute, mlir::Attribute> &bindings,
    llvm::SmallVector<mlir::Diagnostic> &diagnostics
);

mlir::LogicalResult substituteFunctionBody(
    function::FuncDefOp clone, const llvm::DenseMap<mlir::Attribute, mlir::Attribute> &bindings,
    llvm::SmallVector<mlir::Diagnostic> &diagnostics
);
 
} // namespace llzk::polymorphic::detail