99 lines · plain
1// RUN: fir-opt --split-input-file --constant-argument-globalisation-opt < %s | FileCheck %s2 3module {4// Test for "two conditional writes to the same alloca doesn't get replaced."5 func.func @func(%arg0: i32, %arg1: i1) {6 %c2_i32 = arith.constant 2 : i327 %addr = fir.alloca i32 {adapt.valuebyref}8 fir.if %arg1 {9 fir.store %c2_i32 to %addr : !fir.ref<i32>10 } else {11 fir.store %arg0 to %addr : !fir.ref<i32>12 }13 fir.call @sub2(%addr) : (!fir.ref<i32>) -> ()14 return15 }16 func.func private @sub2(!fir.ref<i32>)17 18// CHECK-LABEL: func.func @func19// CHECK-SAME: [[ARG0:%.*]]: i3220// CHECK-SAME: [[ARG1:%.*]]: i1)21// CHECK: [[CONST:%.*]] = arith.constant22// CHECK: [[ADDR:%.*]] = fir.alloca i3223// CHECK: fir.if [[ARG1]]24// CHECK: fir.store [[CONST]] to [[ADDR]]25// CHECK: } else {26// CHECK: fir.store [[ARG0]] to [[ADDR]]27// CHECK: fir.call @sub2([[ADDR]])28// CHECK: return29 30}31 32// -----33 34module {35// Test for "two writes to the same alloca doesn't get replaced."36 func.func @func() {37 %c1_i32 = arith.constant 1 : i3238 %c2_i32 = arith.constant 2 : i3239 %addr = fir.alloca i32 {adapt.valuebyref}40 fir.store %c1_i32 to %addr : !fir.ref<i32>41 fir.store %c2_i32 to %addr : !fir.ref<i32>42 fir.call @sub2(%addr) : (!fir.ref<i32>) -> ()43 return44 }45 func.func private @sub2(!fir.ref<i32>)46 47// CHECK-LABEL: func.func @func48// CHECK: [[CONST1:%.*]] = arith.constant49// CHECK: [[CONST2:%.*]] = arith.constant50// CHECK: [[ADDR:%.*]] = fir.alloca i3251// CHECK: fir.store [[CONST1]] to [[ADDR]]52// CHECK: fir.store [[CONST2]] to [[ADDR]]53// CHECK: fir.call @sub2([[ADDR]])54// CHECK: return55 56}57 58// -----59 60module {61// Test for "one write to the the alloca gets replaced."62 func.func @func() {63 %c1_i32 = arith.constant 1 : i3264 %addr = fir.alloca i32 {adapt.valuebyref}65 fir.store %c1_i32 to %addr : !fir.ref<i32>66 fir.call @sub2(%addr) : (!fir.ref<i32>) -> ()67 return68 }69 func.func private @sub2(!fir.ref<i32>)70 71// CHECK-LABEL: func.func @func72// CHECK: [[ADDR:%.*]] = fir.address_of([[EXTR:@.*]]) : !fir.ref<i32>73// CHECK: fir.call @sub2([[ADDR]])74// CHECK: return75// CHECK: fir.global internal [[EXTR]] constant : i32 {76// CHECK: %{{.*}} = arith.constant 1 : i3277// CHECK: fir.has_value %{{.*}} : i3278// CHECK: }79 80}81 82// -----83// Check that same argument used twice is converted.84module {85 func.func @func(%arg0: !fir.ref<i32>, %arg1: i1) {86 %c2_i32 = arith.constant 2 : i3287 %addr1 = fir.alloca i32 {adapt.valuebyref}88 fir.store %c2_i32 to %addr1 : !fir.ref<i32>89 fir.call @sub1(%addr1, %addr1) : (!fir.ref<i32>, !fir.ref<i32>) -> ()90 return91 }92}93 94// CHECK-LABEL: func.func @func95// CHECK-NEXT: %[[ARG1:.*]] = fir.address_of([[CONST1:@.*]]) : !fir.ref<i32>96// CHECK-NEXT: %[[ARG2:.*]] = fir.address_of([[CONST2:@.*]]) : !fir.ref<i32>97// CHECK-NEXT: fir.call @sub1(%[[ARG1]], %[[ARG2]])98// CHECK-NEXT: return99