StructSpecializationDiscovery.h

Go to the documentation of this file.

//===-- StructSpecializationDiscovery.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 <llvm/ADT/DenseMap.h>
#include <llvm/ADT/SmallVector.h>
 
#include <cstdint>
 
namespace llzk::polymorphic::detail {

mlir::FailureOr<mlir::Attribute>
resolveConstReadBinding(ConstReadOp read, mlir::Attribute binding, mlir::Type readType);

struct StructSpecializationRequest {
  enum class Kind : std::uint8_t { Plain, ArrayElement, RolledCallResult };
 
  mlir::Operation *site;
  component::StructType type;
  llvm::SmallVector<int64_t> arrayIndices;
  component::StructType familyType;
  Kind kind;
};

struct RolledCallTargets {
  mlir::Operation *site;
  llvm::SmallVector<component::StructType> candidates;
};

class StructSpecializationDiscovery {
public:
  using Bindings = llvm::DenseMap<mlir::Attribute, mlir::Attribute>;
  using Requests = llvm::SmallVector<StructSpecializationRequest>;
  using CallTargets = llvm::SmallVector<RolledCallTargets>;

  mlir::FailureOr<Bindings>
  evaluateBindings(mlir::Operation *source, const Bindings &bindings) const;

  mlir::FailureOr<Requests> discover(
      mlir::Operation *source, const Bindings &bindings,
      llvm::SmallVectorImpl<mlir::Operation *> *visitedOperations = nullptr,
      CallTargets *callTargets = nullptr
  ) const;
};
 
} // namespace llzk::polymorphic::detail