1// REQUIRES: asserts 2// RUN: mlir-opt %s --pass-pipeline='builtin.module(func.func(mem2reg))' --split-input-file --mlir-pass-statistics 2>&1 >/dev/null | FileCheck %s 3 4// CHECK: Mem2Reg 5// CHECK-NEXT: (S) 0 new block args 6// CHECK-NEXT: (S) 1 promoted slots 7func.func @basic() -> i32 { 8 %0 = arith.constant 5 : i32 9 %1 = memref.alloca() : memref<i32> 10 memref.store %0, %1[] : memref<i32> 11 %2 = memref.load %1[] : memref<i32> 12 return %2 : i32 13} 14 15// ----- 16 17// CHECK: Mem2Reg 18// CHECK-NEXT: (S) 0 new block args 19// CHECK-NEXT: (S) 0 promoted slots 20func.func @no_alloca() -> i32 { 21 %0 = arith.constant 5 : i32 22 return %0 : i32 23} 24 25// ----- 26 27// CHECK: Mem2Reg 28// CHECK-NEXT: (S) 2 new block args 29// CHECK-NEXT: (S) 1 promoted slots 30func.func @cycle(%arg0: i64, %arg1: i1, %arg2: i64) { 31 %alloca = memref.alloca() : memref<i64> 32 memref.store %arg2, %alloca[] : memref<i64> 33 cf.cond_br %arg1, ^bb1, ^bb2 34^bb1: 35 %use = memref.load %alloca[] : memref<i64> 36 call @use(%use) : (i64) -> () 37 memref.store %arg0, %alloca[] : memref<i64> 38 cf.br ^bb2 39^bb2: 40 cf.br ^bb1 41} 42 43func.func @use(%arg: i64) { return } 44 45// ----- 46 47// CHECK: Mem2Reg 48// CHECK-NEXT: (S) 0 new block args 49// CHECK-NEXT: (S) 3 promoted slots 50func.func @recursive(%arg: i64) -> i64 { 51 %alloca0 = memref.alloca() : memref<memref<memref<i64>>> 52 %alloca1 = memref.alloca() : memref<memref<i64>> 53 %alloca2 = memref.alloca() : memref<i64> 54 memref.store %arg, %alloca2[] : memref<i64> 55 memref.store %alloca2, %alloca1[] : memref<memref<i64>> 56 memref.store %alloca1, %alloca0[] : memref<memref<memref<i64>>> 57 %load0 = memref.load %alloca0[] : memref<memref<memref<i64>>> 58 %load1 = memref.load %load0[] : memref<memref<i64>> 59 %load2 = memref.load %load1[] : memref<i64> 60 return %load2 : i64 61} 62