62 lines · plain
1// REQUIRES: asserts2// RUN: mlir-opt %s --pass-pipeline='builtin.module(func.func(mem2reg))' --split-input-file --mlir-pass-statistics 2>&1 >/dev/null | FileCheck %s3 4// CHECK: Mem2Reg5// CHECK-NEXT: (S) 0 new block args6// CHECK-NEXT: (S) 1 promoted slots7func.func @basic() -> i32 {8 %0 = arith.constant 5 : i329 %1 = memref.alloca() : memref<i32>10 memref.store %0, %1[] : memref<i32>11 %2 = memref.load %1[] : memref<i32>12 return %2 : i3213}14 15// -----16 17// CHECK: Mem2Reg18// CHECK-NEXT: (S) 0 new block args19// CHECK-NEXT: (S) 0 promoted slots20func.func @no_alloca() -> i32 {21 %0 = arith.constant 5 : i3222 return %0 : i3223}24 25// -----26 27// CHECK: Mem2Reg28// CHECK-NEXT: (S) 2 new block args29// CHECK-NEXT: (S) 1 promoted slots30func.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, ^bb234^bb1:35 %use = memref.load %alloca[] : memref<i64>36 call @use(%use) : (i64) -> ()37 memref.store %arg0, %alloca[] : memref<i64>38 cf.br ^bb239^bb2:40 cf.br ^bb141}42 43func.func @use(%arg: i64) { return }44 45// -----46 47// CHECK: Mem2Reg48// CHECK-NEXT: (S) 0 new block args49// CHECK-NEXT: (S) 3 promoted slots50func.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 : i6461}62