Author: Congcong Cai Date: 2024-04-01T14:01:03-07:00 New Revision: e0f0c463b53987adab2e1d190c0e89165c126506
URL: https://github.com/llvm/llvm-project/commit/e0f0c463b53987adab2e1d190c0e89165c126506 DIFF: https://github.com/llvm/llvm-project/commit/e0f0c463b53987adab2e1d190c0e89165c126506.diff LOG: [mlir][transform] replace original op to loop ops (#83537) (cherry picked from commit 0597644a6466ae9148b0b41cb8f95d5022e045c2) Added: Modified: mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp mlir/test/Interfaces/TilingInterface/lower-to-loops-using-interface.mlir Removed: ################################################################################ diff --git a/mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp b/mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp index 140bdd1f2db361..be875297fc93ca 100644 --- a/mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp +++ b/mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp @@ -2092,6 +2092,7 @@ DiagnosedSilenceableFailure transform::ConvertToLoopsOp::applyToOne( scf::lowerToLoopsUsingSCFForOp(rewriter, target); if (failed(loops)) return emitDefaultDefiniteFailure(target); + rewriter.eraseOp(target); return DiagnosedSilenceableFailure::success(); } diff --git a/mlir/test/Interfaces/TilingInterface/lower-to-loops-using-interface.mlir b/mlir/test/Interfaces/TilingInterface/lower-to-loops-using-interface.mlir index 7969de0d456bb6..1b2c553b25ded0 100644 --- a/mlir/test/Interfaces/TilingInterface/lower-to-loops-using-interface.mlir +++ b/mlir/test/Interfaces/TilingInterface/lower-to-loops-using-interface.mlir @@ -33,6 +33,7 @@ module attributes {transform.with_named_sequence} { // CHECK: %[[MULF:.+]] = arith.mulf %[[LHS]], %[[RHS]] // CHECK: %[[ADDF:.+]] = arith.addf %[[OUT]], %[[MULF]] // CHECK: memref.store %[[ADDF]], %[[ARG2]][%[[IV0]], %[[IV1]]] +// CHECK-NOT: linalg.matmul ins(%arg0, %arg1 : memref<?x?xf32>, memref<?x?xf32>) // ----- _______________________________________________ llvm-branch-commits mailing list llvm-branch-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits