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-exception 6 // 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 15 using namespace mlir; 16 17 namespace { 18 /// Debug extension of the Transform dialect. This provides operations for 19 /// debugging transform dialect scripts. 20 class DebugExtension 21 : public transform::TransformDialectExtension<DebugExtension> { 22 public: 23 MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(DebugExtension) 24 25 void init() { 26 registerTransformOps< 27 #define GET_OP_LIST 28 #include "mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.cpp.inc" 29 >(); 30 } 31 }; 32 } // namespace 33 34 void mlir::transform::registerDebugExtension(DialectRegistry &dialectRegistry) { 35 dialectRegistry.addExtensions<DebugExtension>(); 36 } 37