c8fb6ee341
This revision adds a new `replaceOpWithIf` hook that replaces uses of an operation that satisfy a given functor. If all uses are replaced, the operation gets erased in a similar manner to `replaceOp`. DialectConversion support will be added in a followup as this requires adjusting how replacements are tracked there. Differential Revision: https://reviews.llvm.org/D94632
16 lines
564 B
MLIR
16 lines
564 B
MLIR
// RUN: mlir-opt -allow-unregistered-dialect -split-input-file -test-pattern-selective-replacement -verify-diagnostics %s | FileCheck %s
|
|
|
|
// Test that operations can be selectively replaced.
|
|
|
|
// CHECK-LABEL: @test1
|
|
// CHECK-SAME: %[[ARG0:.*]]: i32, %[[ARG1:.*]]: i32
|
|
func @test1(%arg0: i32, %arg1 : i32) -> () {
|
|
// CHECK: addi %[[ARG1]], %[[ARG1]]
|
|
// CHECK-NEXT: "test.return"(%[[ARG0]]
|
|
%cast = "test.cast"(%arg0, %arg1) : (i32, i32) -> (i32)
|
|
%non_terminator = addi %cast, %cast : i32
|
|
"test.return"(%cast, %non_terminator) : (i32, i32) -> ()
|
|
}
|
|
|
|
// -----
|