llvm-project

1
// RUN: fir-opt --split-input-file --fir-memref-dataflow-opt %s | FileCheck %s
2

3
// Test that all load-store chains are removed
4

5
func.func @load_store_chain_removal(%arg0: !fir.ref<!fir.array<60xi32>>, %arg1: !fir.ref<!fir.array<60xi32>>, %arg2: !fir.ref<!fir.array<60xi32>>) {
6
  %c1_i64 = arith.constant 1 : i64
7
  %c60 = arith.constant 60 : index
8
  %c0 = arith.constant 0 : index
9
  %c1 = arith.constant 1 : index
10
  %0 = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFf1dcEi"}
11
  %1 = fir.alloca !fir.array<60xi32> {bindc_name = "t1", uniq_name = "_QFf1dcEt1"}
12
  cf.br ^bb1(%c1, %c60 : index, index)
13
^bb1(%2: index, %3: index):  // 2 preds: ^bb0, ^bb2
14
  %4 = arith.cmpi sgt, %3, %c0 : index
15
  cf.cond_br %4, ^bb2, ^bb3
16
^bb2:  // pred: ^bb1
17
  %5 = fir.convert %2 : (index) -> i32
18
  fir.store %5 to %0 : !fir.ref<i32>
19
  %6 = fir.load %0 : !fir.ref<i32>
20
  %7 = fir.convert %6 : (i32) -> i64
21
  %8 = arith.subi %7, %c1_i64 : i64
22
  %9 = fir.coordinate_of %arg0, %8 : (!fir.ref<!fir.array<60xi32>>, i64) -> !fir.ref<i32>
23
  %10 = fir.load %9 : !fir.ref<i32>
24
  %11 = arith.addi %10, %10 : i32
25
  %12 = fir.coordinate_of %1, %8 : (!fir.ref<!fir.array<60xi32>>, i64) -> !fir.ref<i32>
26
  fir.store %11 to %12 : !fir.ref<i32>
27
  %13 = arith.addi %2, %c1 : index
28
  %14 = arith.subi %3, %c1 : index
29
  cf.br ^bb1(%13, %14 : index, index)
30
^bb3:  // pred: ^bb1
31
  %15 = fir.convert %2 : (index) -> i32
32
  fir.store %15 to %0 : !fir.ref<i32>
33
  cf.br ^bb4(%c1, %c60 : index, index)
34
^bb4(%16: index, %17: index):  // 2 preds: ^bb3, ^bb5
35
  %18 = arith.cmpi sgt, %17, %c0 : index
36
  cf.cond_br %18, ^bb5, ^bb6
37
^bb5:  // pred: ^bb4
38
  %19 = fir.convert %16 : (index) -> i32
39
  fir.store %19 to %0 : !fir.ref<i32>
40
  %20 = fir.load %0 : !fir.ref<i32>
41
  %21 = fir.convert %20 : (i32) -> i64
42
  %22 = arith.subi %21, %c1_i64 : i64
43
  %23 = fir.coordinate_of %1, %22 : (!fir.ref<!fir.array<60xi32>>, i64) -> !fir.ref<i32>
44
  %24 = fir.load %23 : !fir.ref<i32>
45
  %25 = fir.coordinate_of %arg1, %22 : (!fir.ref<!fir.array<60xi32>>, i64) -> !fir.ref<i32>
46
  %26 = fir.load %25 : !fir.ref<i32>
47
  %27 = arith.muli %24, %26 : i32
48
  %28 = fir.coordinate_of %arg2, %22 : (!fir.ref<!fir.array<60xi32>>, i64) -> !fir.ref<i32>
49
  fir.store %27 to %28 : !fir.ref<i32>
50
  %29 = arith.addi %16, %c1 : index
51
  %30 = arith.subi %17, %c1 : index
52
  cf.br ^bb4(%29, %30 : index, index)
53
^bb6:  // pred: ^bb4
54
  %31 = fir.convert %16 : (index) -> i32
55
  fir.store %31 to %0 : !fir.ref<i32>
56
  return
57
}
58

59
// CHECK-LABEL: func @load_store_chain_removal
60
// CHECK-LABEL: ^bb1
61
// CHECK-LABEL: ^bb2:
62
// Make sure the previous fir.store/fir.load pair have been elimated and we 
63
// preserve the last pair of fir.load/fir.store.
64
// CHECK-COUNT-1: %{{.*}} = fir.load %{{.*}} : !fir.ref<i32>
65
// CHECK-COUNT-1: fir.store %{{.*}} to %{{.*}} : !fir.ref<i32>
66
// CHECK-LABEL: ^bb3:
67
// Make sure the fir.store has been removed.
68
// CHECK-NOT:     fir.store %{{.*}} to %{{.*}} : !fir.ref<i32>
69
// CHECK-LABEL: ^bb5:
70
// CHECK:         %{{.*}} = fir.convert %{{.*}} : (index) -> i32
71
// Check that the fir.store/fir.load pair has been removed between the convert.
72
// CHECK-NOT:     fir.store %{{.*}} to %{{.*}} : !fir.ref<i32>
73
// CHECK-NOT:     %{{.*}} = fir.load %{{.*}} : !fir.ref<i32>
74
// CHECK:         %{{.*}} = fir.convert %{{.*}} : (i32) -> i64
75
// CHECK:         %{{.*}} = fir.load %{{.*}} : !fir.ref<i32>
76
// CHECK:         %{{.*}} = fir.load %{{.*}} : !fir.ref<i32>
77
// CHECK:         fir.store %{{.*}} to %{{.*}} : !fir.ref<i32>
78
// CHECK-LABEL: ^bb6:
79
// CHECK-NOT:     fir.store %{{.*}} to %{{.*}} : !fir.ref<i32>
80