[Codegen] Move ErrorCheckingTrackingListener out of iree-dialects (#24466)

Relocate ErrorCheckingTrackingListener into Codegen/Common (namespace mlir::iree_compiler) and repoint its users in the Common and LLVMGPU transform extensions.

Drop the StructuredTransformOpsExtension registration and includes from CommonDialectRegistration and Interfaces: its transform ops (transform.iree.register_match_callbacks/match_callback/take_first/emit_remark) have no production users and their tests were removed in #24500. Build deps are updated accordingly.

Progress toward retiring the iree-dialects dependency.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: Alex-Wengg

compiler/src/iree/compiler/Codegen/Common/BUILD.bazel

@@ -103,6 +103,7 @@ iree_compiler_cc_library(
         "EncodingUtils.cpp",
         "EraseDeadAllocAndStores.cpp",
         "EraseHALDescriptorTypeFromMemRef.cpp",
+        "ErrorCheckingTrackingListener.cpp",
         "FastMathPatterns.cpp",
         "FissionTransferOpsInControlFlow.cpp",
         "FlattenMemRefSubspan.cpp",
@@ -190,6 +191,7 @@ iree_compiler_cc_library(
         "CombineLayoutTransformation.h",
         "EmulateNarrowType.h",
         "EncodingUtils.h",
+        "ErrorCheckingTrackingListener.h",
         "FastMathPatterns.h",
         "PassUtils.h",
         "Passes.h",
@@ -234,7 +236,6 @@ iree_compiler_cc_library(
         "//compiler/src/iree/compiler/Dialect/Util/IR",
         "//compiler/src/iree/compiler/Dialect/Util/Transforms",
         "//compiler/src/iree/compiler/Utils",
-        "//llvm-external-projects/iree-dialects:IREELinalgTransformDialect",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:AMDGPUDialect",
         "@llvm-project//mlir:AMDGPUTransforms",
@@ -331,7 +332,6 @@ iree_compiler_cc_library(
         "//compiler/src/iree/compiler/Dialect/LinalgExt/IR",
         "//compiler/src/iree/compiler/Dialect/LinalgExt/TransformExtensions:LinalgExtExtensions",
         "//compiler/src/iree/compiler/Dialect/TensorExt/IR",
-        "//llvm-external-projects/iree-dialects:IREELinalgTransformDialect",
         "//compiler/src/iree/compiler/Codegen/Dialect/VectorExt/IR:IREEVectorExtDialect",
         "@llvm-project//mlir:AffineDialect",
         "@llvm-project//mlir:AffineUtils",
@@ -376,7 +376,6 @@ iree_compiler_cc_library(
         "@llvm-project//mlir:Support",
         "@llvm-project//mlir:DialectUtils",
         # TransformExtensions (needed for registration in the pass)
-        "//llvm-external-projects/iree-dialects:IREEDialectsTransforms",
         "//compiler/src/iree/compiler/Codegen/Common/TransformExtensions:CommonExtensions",
         "//compiler/src/iree/compiler/Codegen/LLVMCPU/TransformExtensions:LLVMCPUExtensions",
         "//compiler/src/iree/compiler/Codegen/LLVMGPU/TransformExtensions:LLVMGPUExtensions",

compiler/src/iree/compiler/Codegen/Common/CMakeLists.txt

@@ -56,6 +56,7 @@ iree_cc_library(
     "CombineLayoutTransformation.h"
     "EmulateNarrowType.h"
     "EncodingUtils.h"
+    "ErrorCheckingTrackingListener.h"
     "FastMathPatterns.h"
     "PassUtils.h"
     "Passes.h"
@@ -96,6 +97,7 @@ iree_cc_library(
     "EncodingUtils.cpp"
     "EraseDeadAllocAndStores.cpp"
     "EraseHALDescriptorTypeFromMemRef.cpp"
+    "ErrorCheckingTrackingListener.cpp"
     "FastMathPatterns.cpp"
     "FissionTransferOpsInControlFlow.cpp"
     "FlattenMemRefSubspan.cpp"
@@ -180,7 +182,6 @@ iree_cc_library(
   DEPS
     ::PassHeaders
     ::PassesIncGen
-    IREELinalgTransformDialect
     LLVMSupport
     MLIRAMDGPUDialect
     MLIRAMDGPUTransforms
@@ -286,8 +287,6 @@ iree_cc_library(
     ::Common
     ::PassHeaders
     ::PassesIncGen
-    IREEDialectsTransforms
-    IREELinalgTransformDialect
     LLVMSupport
     MLIRAMDGPUTransforms
     MLIRAffineDialect

compiler/src/iree/compiler/Codegen/Common/CommonDialectRegistration.cpp

@@ -4,7 +4,6 @@
 // See https://llvm.org/LICENSE.txt for license information.
 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
-#include "iree-dialects/Dialect/LinalgTransform/StructuredTransformOpsExt.h"
 #include "iree/compiler/Codegen/Common/TransformExtensions/CommonExtensions.h"
 #include "iree/compiler/Codegen/Dialect/Codegen/IR/IREECodegenDialect.h"
 #include "iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUDialect.h"
@@ -97,8 +96,7 @@ void registerTransformDialectTranslationDependentDialects(
   vector::registerBufferizableOpInterfaceExternalModels(registry);
 
   registry.addExtensions<
-      mlir::iree_compiler::IREE::LinalgExt::LinalgExtTransformOpsExtension,
-      transform_ext::StructuredTransformOpsExtension>();
+      mlir::iree_compiler::IREE::LinalgExt::LinalgExtTransformOpsExtension>();
   iree_compiler::registerTransformDialectCommonExtension(registry);
   iree_compiler::registerTransformDialectLLVMCPUExtension(registry);
   iree_compiler::registerTransformDialectLLVMGPUExtension(registry);

compiler/src/iree/compiler/Codegen/Common/ErrorCheckingTrackingListener.cpp

@@ -0,0 +1,40 @@
+// Copyright 2022 The IREE Authors
+//
+// Licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+#include "iree/compiler/Codegen/Common/ErrorCheckingTrackingListener.h"
+
+#include "mlir/Dialect/SCF/IR/SCF.h"
+#include "llvm/Support/Debug.h"
+
+#define DEBUG_TYPE "iree-codegen-error-checking-tracking-listener"
+#define DBGS() (llvm::dbgs() << "[" DEBUG_TYPE "]: ")
+
+namespace mlir::iree_compiler {
+
+void ErrorCheckingTrackingListener::notifyPayloadReplacementNotFound(
+    Operation *op, ValueRange values, DiagnosedSilenceableFailure &&diag) {
+  // Certain ops can dropped safely.
+  if (isa<scf::ForOp>(op)) {
+    LLVM_DEBUG(DBGS() << "Silently dropping scf.for op mapping\n");
+    return;
+  }
+
+  SmallVector<Diagnostic> diags;
+  diag.takeDiagnostics(diags);
+  if (!status.succeeded()) {
+    status.takeDiagnostics(diags);
+  }
+  status = DiagnosedSilenceableFailure::silenceableFailure(std::move(diags));
+
+  status = emitSilenceableFailure(
+      getTransformOp(), "!!! tracking listener failed to find replacement op");
+  status.attachNote(op->getLoc()) << "replaced op";
+  for (Value v : values) {
+    status.attachNote(v.getLoc()) << "replacement value";
+  }
+}
+
+} // namespace mlir::iree_compiler

compiler/src/iree/compiler/Codegen/Common/ErrorCheckingTrackingListener.h

@@ -0,0 +1,48 @@
+// Copyright 2022 The IREE Authors
+//
+// Licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+#ifndef IREE_COMPILER_CODEGEN_COMMON_ERRORCHECKINGTRACKINGLISTENER_H_
+#define IREE_COMPILER_CODEGEN_COMMON_ERRORCHECKINGTRACKINGLISTENER_H_
+
+#include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h"
+
+namespace mlir::iree_compiler {
+
+/// A tracking listener for tensor IR that checks for payload replacement
+/// errors.
+class ErrorCheckingTrackingListener : public transform::TrackingListener {
+public:
+  using transform::TrackingListener::TrackingListener;
+
+  ~ErrorCheckingTrackingListener() override {
+    assert(status.succeeded() && "must check listener error state");
+  }
+
+  /// Return "true" if this tracking listener had a failure.
+  bool failed() const { return !status.succeeded(); }
+
+  /// Check and return the current error state of this listener. In case of a
+  /// failure state, only the most recent error is returned. Afterwards, resets
+  /// the error state.
+  DiagnosedSilenceableFailure checkAndResetError() {
+    DiagnosedSilenceableFailure result(std::move(status));
+    status = DiagnosedSilenceableFailure::success();
+    return result;
+  }
+
+private:
+  void
+  notifyPayloadReplacementNotFound(Operation *op, ValueRange values,
+                                   DiagnosedSilenceableFailure &&diag) override;
+
+  /// The error state of this listener. "Success" indicates that no error
+  /// happened so far. Otherwise, the status contains the most recent error.
+  DiagnosedSilenceableFailure status = DiagnosedSilenceableFailure::success();
+};
+
+} // namespace mlir::iree_compiler
+
+#endif // IREE_COMPILER_CODEGEN_COMMON_ERRORCHECKINGTRACKINGLISTENER_H_

compiler/src/iree/compiler/Codegen/Common/TransformExtensions/BUILD.bazel

@@ -75,8 +75,6 @@ iree_compiler_cc_library(
         "//compiler/src/iree/compiler/Codegen/Utils",
         "//compiler/src/iree/compiler/Dialect/HAL/IR",
         "//compiler/src/iree/compiler/Dialect/LinalgExt/IR",
-        "//llvm-external-projects/iree-dialects:IREEDialectsTransforms",
-        "//llvm-external-projects/iree-dialects:IREELinalgTransformDialect",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:AffineDialect",
         "@llvm-project//mlir:AffineUtils",

compiler/src/iree/compiler/Codegen/Common/TransformExtensions/CMakeLists.txt

@@ -31,8 +31,6 @@ iree_cc_library(
     "CommonExtensionsOps.cpp.inc"
   DEPS
     ::CommonExtensionsOpGen
-    IREEDialectsTransforms
-    IREELinalgTransformDialect
     LLVMSupport
     MLIRAffineDialect
     MLIRAffineUtils

compiler/src/iree/compiler/Codegen/Common/TransformExtensions/CommonExtensions.cpp

@@ -6,8 +6,7 @@
 
 #include "CommonExtensions.h"
 
-#include "iree-dialects/Dialect/LinalgTransform/StructuredTransformOpsExt.h"
-#include "iree-dialects/Transforms/TransformMatchers.h"
+#include "iree/compiler/Codegen/Common/ErrorCheckingTrackingListener.h"
 #include "iree/compiler/Codegen/Common/GPU/GPUPatterns.h"
 #include "iree/compiler/Codegen/Common/GPU/GPUVectorDistribution.h"
 #include "iree/compiler/Codegen/Common/GPU/Passes.h"
@@ -48,6 +47,7 @@
 #include "mlir/Dialect/SCF/Utils/Utils.h"
 #include "mlir/Dialect/Tensor/IR/Tensor.h"
 #include "mlir/Dialect/Tensor/Transforms/Transforms.h"
+#include "mlir/Dialect/Transform/IR/TransformTypes.h"
 #include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h"
 #include "mlir/Dialect/Utils/StaticValueUtils.h"
 #include "mlir/Dialect/Vector/IR/VectorOps.h"

compiler/src/iree/compiler/Codegen/Interfaces/BUILD.bazel

@@ -61,7 +61,6 @@ iree_compiler_cc_library(
         "//compiler/src/iree/compiler/Codegen/Dialect/PCF/ExternalInterfaces:ExternalModels",
         "//compiler/src/iree/compiler/Codegen/ExternalInterfaces:ExternalModels",
         "//compiler/src/iree/compiler/Dialect/LinalgExt/TransformExtensions:LinalgExtExtensions",
-        "//llvm-external-projects/iree-dialects:IREELinalgTransformDialect",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:LinalgTransforms",
         "@llvm-project//mlir:SideEffectInterfaces",

compiler/src/iree/compiler/Codegen/Interfaces/CMakeLists.txt

@@ -25,7 +25,6 @@ iree_cc_library(
     ::TensorMaskingOpInterface
     ::UKernelOpInterface
     ::VectorizableOpInterface
-    IREELinalgTransformDialect
     MLIRAMDGPUDialect
     MLIRAffineDialect
     MLIRAffineTransformOps