mlir/test/IR/greedy-pattern-rewrite-driver-top-down.mlir

   1 // RUN: mlir-opt %s -test-greedy-patterns="max-iterations=1 top-down=true" \
   2 // RUN:     --split-input-file | FileCheck %s
   3
   4 // Tests for https://github.com/llvm/llvm-project/issues/86765. Ensure
   5 // that operands of a dead op are added to the worklist even if the same value
   6 // appears multiple times as an operand.
   7
   8 // 2 uses of the same operand
   9
  10 // CHECK:       func.func @f(%arg0: i1) {
  11 // CHECK-NEXT:    return
  12 // CHECK-NEXT:  }
  13 func.func @f(%arg0: i1) {
  14   %0 = arith.constant 0 : i32
  15   %if = scf.if %arg0 -> (i32) {
  16     scf.yield %0 : i32
  17   } else {
  18     scf.yield %0 : i32
  19   }
  20   %dead_leaf = arith.addi %if, %if : i32
  21   return
  22 }
  23
  24 // -----
  25
  26 // 3 uses of the same operand
  27
  28 // CHECK:       func.func @f() {
  29 // CHECK-NEXT:    return
  30 // CHECK-NEXT:  }
  31 func.func @f() {
  32   %0 = arith.constant 0 : i1
  33   %if = scf.if %0 -> (i1) {
  34     scf.yield %0 : i1
  35   } else {
  36     scf.yield %0 : i1
  37   }
  38   %dead_leaf = arith.select %if, %if, %if : i1
  39   return
  40 }
  41
  42 // -----
  43
  44 // 2 uses of the same operand, op has 3 operands
  45
  46 // CHECK:       func.func @f() {
  47 // CHECK-NEXT:    return
  48 // CHECK-NEXT:  }
  49 func.func @f() {
  50   %0 = arith.constant 0 : i1
  51   %if = scf.if %0 -> (i1) {
  52     scf.yield %0 : i1
  53   } else {
  54     scf.yield %0 : i1
  55   }
  56   %dead_leaf = arith.select %0, %if, %if : i1
  57   return
  58 }