37 lines · cpp
1//===- DebugExtension.cpp - Debug 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/DebugExtension/DebugExtension.h"10 11#include "mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.h"12#include "mlir/Dialect/Transform/IR/TransformDialect.h"13#include "mlir/IR/DialectRegistry.h"14 15using namespace mlir;16 17namespace {18/// Debug extension of the Transform dialect. This provides operations for19/// debugging transform dialect scripts.20class DebugExtension21 : public transform::TransformDialectExtension<DebugExtension> {22public:23 MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(DebugExtension)24 25 void init() {26 registerTransformOps<27#define GET_OP_LIST28#include "mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.cpp.inc"29 >();30 }31};32} // namespace33 34void mlir::transform::registerDebugExtension(DialectRegistry &dialectRegistry) {35 dialectRegistry.addExtensions<DebugExtension>();36}37