1
1
// RUN: mlir-opt --split-input-file -verify-diagnostics --tosa-layerwise-constant-fold %s | FileCheck %s
2
2
3
- // -----
4
3
// Float additions
5
4
6
5
// CHECK-LABEL: @add_fold_float
7
6
func.func @add_fold_float () -> tensor <4 xf16 > {
8
7
// CHECK: [[RES:]] ={{.*}}tosa.const{{.*}}-1.5{{.*}}e+02, 1.9{{.*}}e+00, 0.{{0*}}e+00, 5.{{0*}}e+00
9
- // CHECK-NOT: tosa.cast
8
+ // CHECK-NOT: tosa.add
10
9
// CHECK: return [[RES]]
11
10
%0 = " tosa.const" () {value =
12
11
dense <[-17.4978 , 4.9882 , 0.0 , -0.0 ]> :
@@ -23,7 +22,7 @@ func.func @add_fold_float() -> tensor<4xf16> {
23
22
// CHECK-LABEL: @add_fold_float_infinity_nan
24
23
func.func @add_fold_float_infinity_nan () -> tensor <6 xf32 > {
25
24
// CHECK: [[RES:]] ={{.*}}tosa.const{{.*}}0x7F800000, 0xFF800000, 0x7F800000, 0xFF800000, 0x7FC00000, 0x7FC00000
26
- // CHECK-NOT: tosa.cast
25
+ // CHECK-NOT: tosa.add
27
26
// CHECK: return [[RES]]
28
27
%0 = " tosa.const" () {value =
29
28
dense <[0x7F800000 , 0xFF800000 , 0x7F800000 , 0xFF800000 , 0x7FC00000 , 0x7F800000 ]> :
@@ -40,7 +39,7 @@ func.func @add_fold_float_infinity_nan() -> tensor<6xf32> {
40
39
// CHECK-LABEL: @add_fold_float_overflow
41
40
func.func @add_fold_float_overflow () -> tensor <2 xf32 > {
42
41
// CHECK: [[RES:]] ={{.*}}tosa.const{{.*}}0x7F800000, 0xFF800000
43
- // CHECK-NOT: tosa.cast
42
+ // CHECK-NOT: tosa.add
44
43
// CHECK: return [[RES]]
45
44
%0 = " tosa.const" () {value =
46
45
dense <[3.1e+38 , -3.1e+38 ]> :
@@ -60,7 +59,7 @@ func.func @add_fold_float_overflow() -> tensor<2xf32> {
60
59
// CHECK-LABEL: @add_fold_int
61
60
func.func @add_fold_int () -> tensor <4 xi32 > {
62
61
// CHECK: [[RES:]] ={{.*}}tosa.const{{.*}}-149, 1, 0, 5
63
- // CHECK-NOT: tosa.cast
62
+ // CHECK-NOT: tosa.add
64
63
// CHECK: return [[RES]]
65
64
%0 = " tosa.const" () {value =
66
65
dense <[-17 , 4 , 0 , 0 ]> :
@@ -77,7 +76,7 @@ func.func @add_fold_int() -> tensor<4xi32> {
77
76
// CHECK-LABEL: @add_fold_int_overflow
78
77
func.func @add_fold_int_overflow () -> tensor <4 xi32 > {
79
78
// CHECK: [[RES:]] ={{.*}}tosa.const{{.*}}2147483647, 2147483647, -2147483648, -2147483648
80
- // CHECK-NOT: tosa.cast
79
+ // CHECK-NOT: tosa.add
81
80
// CHECK: return [[RES]]
82
81
%0 = " tosa.const" () {value =
83
82
dense <[2147483647 , 2147483640 , -2147483648 , -2147483640 ]> :
@@ -97,7 +96,7 @@ func.func @add_fold_int_overflow() -> tensor<4xi32> {
97
96
// CHECK-LABEL: @add_fold_equal_args
98
97
func.func @add_fold_equal_args () -> tensor <3 xi32 > {
99
98
// CHECK: [[RES:]] ={{.*}}tosa.const{{.*}}-34, 8, 0
100
- // CHECK-NOT: tosa.cast
99
+ // CHECK-NOT: tosa.add
101
100
// CHECK: return [[RES]]
102
101
%0 = " tosa.const" () {value =
103
102
dense <[-17 , 4 , 0 ]> :
@@ -113,7 +112,7 @@ func.func @add_fold_equal_args() -> tensor<3xi32> {
113
112
// CHECK-LABEL: @add_fold_int_broadcast_simple
114
113
func.func @add_fold_int_broadcast_simple () -> tensor <3 xi32 > {
115
114
// CHECK: [[RES:]] ={{.*}}tosa.const{{.*}}-29, -8, -12
116
- // CHECK-NOT: tosa.cast
115
+ // CHECK-NOT: tosa.add
117
116
// CHECK: return [[RES]]
118
117
%0 = " tosa.const" () {value =
119
118
dense <[-17 , 4 , 0 ]> :
@@ -133,7 +132,7 @@ func.func @add_fold_int_broadcast_complex() -> tensor<3x3xi32> {
133
132
// CHECK-SAME{LITERAL}: [[-29, -10, -13],
134
133
// CHECK-SAME{LITERAL}: [-11, 8, 5],
135
134
// CHECK-SAME{LITERAL}: [7, 26, 23]]
136
- // CHECK-NOT: tosa.cast
135
+ // CHECK-NOT: tosa.add
137
136
// CHECK: return [[RES]]
138
137
%0 = " tosa.const" () {value =
139
138
dense <[[-17 ], [1 ], [19 ]]> :
0 commit comments