OpTraits.cpp

Go to the documentation of this file.

//===-- OpTraits.cpp --------------------------------------------*- 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
//
//===----------------------------------------------------------------------===//
 
#include "llzk/Dialect/Function/IR/OpTraits.h"
 
#include "llzk/Dialect/Function/IR/Ops.h"
#include "llzk/Dialect/Polymorphic/IR/Ops.h"
#include "llzk/Dialect/Shared/OpHelpers.h"
#include "llzk/Dialect/Verif/IR/Ops.h"
 
#include <mlir/IR/Operation.h>
 
using namespace mlir;
 
namespace llzk::function {
 
namespace {
 
auto parentFuncDefOpHasAttr = [](Operation *op, auto attrFn) -> bool {
  if (FuncDefOp f = op->getParentOfType<FuncDefOp>()) {
    return (f.*attrFn)();
  }
  return false;
};
 
} // namespace
 
LogicalResult verifyConstraintGenTraitImpl(Operation *op) {
  if (parentFuncDefOpHasAttr(op, &FuncDefOp::hasAllowConstraintAttr)) {
    return success();
  }
  return op->emitOpError() << "only valid within a '" << FuncDefOp::getOperationName() << "' with '"
                           << AllowConstraintAttr::name << "' attribute";
}
 
LogicalResult verifyWitnessGenTraitImpl(Operation *op) {
  if (parentFuncDefOpHasAttr(op, &FuncDefOp::hasAllowWitnessAttr)) {
    return success();
  }
  return op->emitOpError() << "only valid within a '" << FuncDefOp::getOperationName() << "' with '"
                           << AllowWitnessAttr::name << "' attribute";
}
 
LogicalResult verifyNotFieldNativeTraitImpl(Operation *op) {
  // These are allowed anywhere outside of FuncDefOp but only allowed inside a FuncDefOp
  // that is marked with the associated attribute.
  if (FuncDefOp f = op->getParentOfType<FuncDefOp>()) {
    if (!f.hasAllowNonNativeFieldOpsAttr()) {
      return op->emitOpError() << "cannot be used within a '" << FuncDefOp::getOperationName()
                               << "' without the '" << AllowNonNativeFieldOpsAttr::name
                               << "' attribute";
    }
  }
  return success();
}
 
} // namespace llzk::function