-
Notifications
You must be signed in to change notification settings - Fork 10.7k
/
ops.mlir
87 lines (59 loc) · 2.84 KB
/
ops.mlir
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
// RUN: mlir-opt %s | mlir-opt | FileCheck %s
// RUN: mlir-opt %s --mlir-print-op-generic | mlir-opt | FileCheck %s
// CHECK-LABEL: func @ops(
// CHECK-SAME: %[[F:.*]]: f32) {
func.func @ops(%f: f32) {
// CHECK: complex.constant [1.{{.*}}, -1.{{.*}}] : complex<f64>
%cst_f64 = complex.constant [0.1, -1.0] : complex<f64>
// CHECK: complex.constant [1.{{.*}} : f32, -1.{{.*}} : f32] : complex<f32>
%cst_f32 = complex.constant [0.1 : f32, -1.0 : f32] : complex<f32>
// CHECK: %[[C:.*]] = complex.create %[[F]], %[[F]] : complex<f32>
%complex = complex.create %f, %f : complex<f32>
// CHECK: complex.re %[[C]] : complex<f32>
%real = complex.re %complex : complex<f32>
// CHECK: complex.im %[[C]] : complex<f32>
%imag = complex.im %complex : complex<f32>
// CHECK: complex.abs %[[C]] : complex<f32>
%abs = complex.abs %complex : complex<f32>
// CHECK: complex.add %[[C]], %[[C]] : complex<f32>
%sum = complex.add %complex, %complex : complex<f32>
// CHECK: complex.cos %[[C]] : complex<f32>
%cos = complex.cos %complex : complex<f32>
// CHECK: complex.div %[[C]], %[[C]] : complex<f32>
%div = complex.div %complex, %complex : complex<f32>
// CHECK: complex.eq %[[C]], %[[C]] : complex<f32>
%eq = complex.eq %complex, %complex : complex<f32>
// CHECK: complex.exp %[[C]] : complex<f32>
%exp = complex.exp %complex : complex<f32>
// CHECK: complex.expm1 %[[C]] : complex<f32>
%expm1 = complex.expm1 %complex : complex<f32>
// CHECK: complex.log %[[C]] : complex<f32>
%log = complex.log %complex : complex<f32>
// CHECK: complex.log1p %[[C]] : complex<f32>
%log1p = complex.log1p %complex : complex<f32>
// CHECK: complex.mul %[[C]], %[[C]] : complex<f32>
%prod = complex.mul %complex, %complex : complex<f32>
// CHECK: complex.neg %[[C]] : complex<f32>
%neg = complex.neg %complex : complex<f32>
// CHECK: complex.neq %[[C]], %[[C]] : complex<f32>
%neq = complex.neq %complex, %complex : complex<f32>
// CHECK: complex.sign %[[C]] : complex<f32>
%sign = complex.sign %complex : complex<f32>
// CHECK: complex.sin %[[C]] : complex<f32>
%sin = complex.sin %complex : complex<f32>
// CHECK: complex.sub %[[C]], %[[C]] : complex<f32>
%diff = complex.sub %complex, %complex : complex<f32>
// CHECK: complex.tanh %[[C]] : complex<f32>
%tanh = complex.tanh %complex : complex<f32>
// CHECK: complex.pow %[[C]], %[[C]] : complex<f32>
%pow = complex.pow %complex, %complex : complex<f32>
// CHECK: complex.sqrt %[[C]] : complex<f32>
%sqrt = complex.sqrt %complex : complex<f32>
// CHECK: complex.rsqrt %[[C]] : complex<f32>
%rsqrt = complex.rsqrt %complex : complex<f32>
// CHECK: complex.atan2 %[[C]], %[[C]] : complex<f32>
%atan2 = complex.atan2 %complex, %complex : complex<f32>
// CHECK: complex.tan %[[C]] : complex<f32>
%tan = complex.tan %complex : complex<f32>
return
}