36 lines · cpp
1//===- SMTExtension.cpp - SMT extension for the Transform dialect ---------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "mlir/Dialect/Transform/SMTExtension/SMTExtension.h"10#include "mlir/Dialect/Transform/SMTExtension/SMTExtensionOps.h"11#include "mlir/IR/DialectRegistry.h"12 13using namespace mlir;14 15//===----------------------------------------------------------------------===//16// Transform op registration17//===----------------------------------------------------------------------===//18 19namespace {20class SMTExtension : public transform::TransformDialectExtension<SMTExtension> {21public:22 MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(SMTExtension)23 24 SMTExtension() {25 registerTransformOps<26#define GET_OP_LIST27#include "mlir/Dialect/Transform/SMTExtension/SMTExtensionOps.cpp.inc"28 >();29 }30};31} // namespace32 33void mlir::transform::registerSMTExtension(DialectRegistry &dialectRegistry) {34 dialectRegistry.addExtensions<SMTExtension>();35}36