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/Function/IR/Attrs.h"
#include "llzk/Dialect/Polymorphic/IR/Ops.h"
#include "llzk/Dialect/Shared/OpHelpers.h"
#include "llzk/Dialect/Struct/IR/Ops.h"
#include "llzk/Util/Constants.h"
#include "llzk/Util/SymbolHelper.h"
 
#include <mlir/IR/OpImplementation.h>
#include <mlir/Interfaces/CallInterfaces.h>
#include <mlir/Interfaces/ControlFlowInterfaces.h>
#include <mlir/Interfaces/FunctionInterfaces.h>
 
#include <cstdint>
#include <optional>
 
// Include TableGen'd declarations
#define GET_OP_CLASSES
#include "llzk/Dialect/Function/IR/Ops.h.inc"
 
namespace llzk::function {

constexpr char ARG_NAME_ATTR_NAME[] = "function.arg_name";

constexpr char RES_NAME_ATTR_NAME[] = "function.res_name";

enum class FunctionKind : std::uint8_t {
  StructCompute,
  StructConstrain,
  StructProduct,
  Free
};

FunctionKind fnNameToKind(mlir::StringRef name);
 
} // namespace llzk::function