1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
// Test regrouping of + and - operations on complex components into complex operations
// RUN: fir-opt --canonicalize %s | FileCheck %s
// CHECK-LABEL: @add
func.func @add(%z: !fir.ref<!fir.complex<8>>, %z1 : !fir.complex<8>, %z2 : !fir.complex<8>) {
%c0 = arith.constant 0 : index
%c1 = arith.constant 1 : index
%real1 = fir.extract_value %z1, [0 : index] : (!fir.complex<8>) -> f64
%imag1 = fir.extract_value %z1, [1 : index] : (!fir.complex<8>) -> f64
%real2 = fir.extract_value %z2, [0 : index] : (!fir.complex<8>) -> f64
%imag2 = fir.extract_value %z2, [1 : index] : (!fir.complex<8>) -> f64
// CHECK-LABEL: fir.addc
%real = arith.addf %real1, %real2 : f64
%imag = arith.addf %imag1, %imag2 : f64
%undef = fir.undefined !fir.complex<8>
%insert_real = fir.insert_value %undef, %real, [0 : index] : (!fir.complex<8>, f64) -> !fir.complex<8>
%insert_imag = fir.insert_value %insert_real, %imag, [1 : index] : (!fir.complex<8>, f64) -> !fir.complex<8>
fir.store %insert_imag to %z : !fir.ref<!fir.complex<8>>
return
}
// CHECK-LABEL: @sub
func.func @sub(%z: !fir.ref<!fir.complex<8>>, %z1 : !fir.complex<8>, %z2 : !fir.complex<8>) {
%c0 = arith.constant 0 : index
%c1 = arith.constant 1 : index
%real1 = fir.extract_value %z1, [0 : index] : (!fir.complex<8>) -> f64
%imag1 = fir.extract_value %z1, [1 : index] : (!fir.complex<8>) -> f64
%real2 = fir.extract_value %z2, [0 : index] : (!fir.complex<8>) -> f64
%imag2 = fir.extract_value %z2, [1 : index] : (!fir.complex<8>) -> f64
// CHECK-LABEL: fir.subc
%real = arith.subf %real1, %real2 : f64
%imag = arith.subf %imag1, %imag2 : f64
%undef = fir.undefined !fir.complex<8>
%insert_real = fir.insert_value %undef, %real, [0 : index] : (!fir.complex<8>, f64) -> !fir.complex<8>
%insert_imag = fir.insert_value %insert_real, %imag, [1 : index] : (!fir.complex<8>, f64) -> !fir.complex<8>
fir.store %insert_imag to %z : !fir.ref<!fir.complex<8>>
return
}
// CHECK-LABEL: @undefOpHiddenByBranch
func.func @undefOpHiddenByBranch(%z: !fir.ref<!fir.complex<8>>, %b: i1) {
%c0 = arith.constant 0 : index
%c1 = arith.constant 1 : index
cf.cond_br %b, ^bb1, ^bb2
^bb1: // pred: ^bb0
%u1 = fir.undefined !fir.complex<8>
%z1l = fir.call @bar1() : () -> !fir.complex<8>
%z1r = fir.call @bar1() : () -> !fir.complex<8>
cf.br ^bb3(%u1, %z1l, %z1r : !fir.complex<8>, !fir.complex<8>, !fir.complex<8>)
^bb2: // pred: ^bb0
%u2 = fir.undefined !fir.complex<8>
%z2l = fir.call @bar2() : () -> !fir.complex<8>
%z2r = fir.call @bar2() : () -> !fir.complex<8>
cf.br ^bb3(%u2, %z2l, %z2r : !fir.complex<8>, !fir.complex<8>, !fir.complex<8>)
// CHECK: ^bb3(%[[z1:.*]]: !fir.complex<8>, %[[z2:.*]]: !fir.complex<8>): // 2 preds: ^bb1, ^bb2
// CHECK: fir.addc %[[z1]], %[[z2]] : !fir.complex<8>
^bb3(%undef : !fir.complex<8>, %z1 : !fir.complex<8>, %z2 : !fir.complex<8>): // 2 preds: ^bb1, ^bb2
%real1 = fir.extract_value %z1, [0 : index] : (!fir.complex<8>) -> f64
%imag1 = fir.extract_value %z1, [1 : index] : (!fir.complex<8>) -> f64
%real2 = fir.extract_value %z2, [0 : index] : (!fir.complex<8>) -> f64
%imag2 = fir.extract_value %z2, [1 : index] : (!fir.complex<8>) -> f64
%real = arith.addf %real1, %real2 : f64
%imag = arith.addf %imag1, %imag2 : f64
%insert_real = fir.insert_value %undef, %real, [0 : index] : (!fir.complex<8>, f64) -> !fir.complex<8>
%insert_imag = fir.insert_value %insert_real, %imag, [1 : index] : (!fir.complex<8>, f64) -> !fir.complex<8>
fir.store %insert_imag to %z : !fir.ref<!fir.complex<8>>
return
}
func.func private @bar1() -> !fir.complex<8>
func.func private @bar2() -> !fir.complex<8>
// CHECK-LABEL: @close_but_bad_pattern
func.func @close_but_bad_pattern(%z: !fir.ref<!fir.complex<8>>, %z1 : !fir.complex<8>, %z2 : !fir.complex<8>) {
%c0 = arith.constant 0 : index
%c1 = arith.constant 1 : index
%real1 = fir.extract_value %z1, [0 : index] : (!fir.complex<8>) -> f64
// extracting %c0 instead of %c1
%imag1 = fir.extract_value %z1, [0 : index] : (!fir.complex<8>) -> f64
%real2 = fir.extract_value %z2, [0 : index] : (!fir.complex<8>) -> f64
%imag2 = fir.extract_value %z2, [1 : index] : (!fir.complex<8>) -> f64
// CHECK: arith.subf
// CHECK: subf
%real = arith.subf %real1, %real2 : f64
%imag = arith.subf %imag1, %imag2 : f64
%undef = fir.undefined !fir.complex<8>
// CHECK: %[[insert1:.*]] = fir.insert_value %{{.*}}, %{{.*}}, [0
// CHECK: %[[insert2:.*]] = fir.insert_value %[[insert1]], %{{.*}}, [1
%insert_real = fir.insert_value %undef, %real, [0 : index] : (!fir.complex<8>, f64) -> !fir.complex<8>
%insert_imag = fir.insert_value %insert_real, %imag, [1 : index] : (!fir.complex<8>, f64) -> !fir.complex<8>
// CHECK: fir.store %[[insert2]] to {{.*}}
fir.store %insert_imag to %z : !fir.ref<!fir.complex<8>>
return
}
|
