Ops.h

Go to the documentation of this file.

//===-- Ops.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/Dialect/Polymorphic/IR/Dialect.h"
#include "llzk/Dialect/Polymorphic/IR/Types.h"
#include "llzk/Dialect/Shared/OpHelpers.h"
 
#include <mlir/Dialect/Affine/IR/AffineValueMap.h>
#include <mlir/Interfaces/ControlFlowInterfaces.h>
#include <mlir/Interfaces/InferTypeOpInterface.h>
 
#include <llvm/ADT/StringRef.h>
 
// Include TableGen'd declarations
#include "llzk/Dialect/Polymorphic/IR/OpInterfaces.h.inc"
 
namespace llzk::polymorphic {

inline constexpr llvm::StringLiteral TEMPLATE_NAME_PATTERN_ATTR = "poly.name_pattern";

inline constexpr llvm::StringLiteral SPECIALIZATION_ID_ATTR = "poly.specialization_id";
inline constexpr llvm::StringLiteral SPECIALIZATION_ORIGIN_ATTR = "poly.origin";
inline constexpr llvm::StringLiteral SPECIALIZATION_ARGUMENTS_ATTR = "poly.arguments";
inline constexpr llvm::StringLiteral FAMILY_ATTR = "poly.family";
inline constexpr llvm::StringLiteral FAMILY_SPECIALIZATIONS_ATTR = "poly.family_specializations";
inline constexpr llvm::StringLiteral FAMILY_INDICES_KEY = "indices";
inline constexpr llvm::StringLiteral FAMILY_SPECIALIZATION_KEY = "specialization";
inline constexpr llvm::StringLiteral FAMILY_TYPE_KEY = "type";
 
template <typename OpT>
concept TemplateSymbolBindingOp =
    std::is_same_v<OpT, TemplateSymbolBindingOpInterface> ||
    std::is_base_of_v<TemplateSymbolBindingOpInterface::Trait<OpT>, OpT>;
} // namespace llzk::polymorphic
 
// Include TableGen'd declarations
#define GET_OP_CLASSES
#include "llzk/Dialect/Polymorphic/IR/Ops.h.inc"
 
namespace llzk::polymorphic {

bool isInTemplate(mlir::Operation *op);

mlir::FailureOr<TemplateOp> verifyInTemplate(mlir::Operation *op);
 
} // namespace llzk::polymorphic