Skip to content

Commit 8eccc9c

Browse files
committed
Automatically upgrade MLIR check tests
All changes in this commit where done automatically with the following Python script: https://gist.github.com/MacDue/ca84d3ec19cf83ae71aab2be8f09c3c5
1 parent ed70312 commit 8eccc9c

33 files changed

+672
-672
lines changed

mlir/test/Conversion/ArithToAMDGPU/8-bit-float-saturation.mlir

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ func.func @scalar_trunc(%v: f16) -> f8E5M2FNUZ {
4646
// CHECK: [[F0:%.+]] = vector.extract [[SATURATED]][0]
4747
// CHECK: [[F1:%.+]] = vector.extract [[SATURATED]][1]
4848
// CHECK: [[W0:%.+]] = amdgpu.packed_trunc_2xfp8 [[F0]], [[F1]] into undef[word 0] : f32 to vector<4xf8E4M3FNUZ>
49-
// CHECK: [[W:%.+]] = vector.extract_strided_slice [[W0]] {offsets = [0], sizes = [2], strides = [1]} : vector<4xf8E4M3FNUZ> to vector<2xf8E4M3FNUZ>
49+
// CHECK: [[W:%.+]] = vector.extract_strided_slice [[W0]][0:2:1] : vector<4xf8E4M3FNUZ> to vector<2xf8E4M3FNUZ>
5050
// CHECK: return [[W]] : vector<2xf8E4M3FNUZ>
5151
func.func @vector_trunc_short(%v: vector<2xf32>) -> vector<2xf8E4M3FNUZ> {
5252
%w = arith.truncf %v : vector<2xf32> to vector<2xf8E4M3FNUZ>

mlir/test/Conversion/ArithToAMDGPU/8-bit-floats.mlir

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ func.func @vector_ext_short(%v: vector<2xf8E5M2FNUZ>) -> vector<2xf64> {
3434

3535
// CHECK-LABEL: func.func @vector_ext_long
3636
// CHECK-SAME: ([[V:%.+]]: vector<9xf8E4M3FNUZ>)
37-
// CHECK: [[V0:%.+]] = vector.extract_strided_slice [[V]] {offsets = [0], sizes = [4], strides = [1]}
37+
// CHECK: [[V0:%.+]] = vector.extract_strided_slice [[V]][0:4:1]
3838
// CHECK: [[F0:%.+]] = amdgpu.ext_packed_fp8 [[V0]][0]
3939
// CHECK: [[W0:%.+]] = vector.insert [[F0]]
4040
// CHECK: [[F1:%.+]] = amdgpu.ext_packed_fp8 [[V0]][1]
@@ -44,7 +44,7 @@ func.func @vector_ext_short(%v: vector<2xf8E5M2FNUZ>) -> vector<2xf64> {
4444
// CHECK: [[F3:%.+]] = amdgpu.ext_packed_fp8 [[V0]][3]
4545
// CHECK: [[W3:%.+]] = vector.insert [[F3]], [[W2]]
4646

47-
// CHECK: [[V1:%.+]] = vector.extract_strided_slice [[V]] {offsets = [4], sizes = [4], strides = [1]} : vector<9xf8E4M3FNUZ> to vector<4xf8E4M3FNUZ>
47+
// CHECK: [[V1:%.+]] = vector.extract_strided_slice [[V]][4:4:1] : vector<9xf8E4M3FNUZ> to vector<4xf8E4M3FNUZ>
4848
// CHECK: [[F4:%.+]] = amdgpu.ext_packed_fp8 [[V1]][0]
4949
// CHECK: [[W4:%.+]] = vector.insert [[F4]], [[W3]]
5050
// CHECK: [[F5:%.+]] = amdgpu.ext_packed_fp8 [[V1]][1]
@@ -54,7 +54,7 @@ func.func @vector_ext_short(%v: vector<2xf8E5M2FNUZ>) -> vector<2xf64> {
5454
// CHECK: [[F7:%.+]] = amdgpu.ext_packed_fp8 [[V1]][3]
5555
// CHECK: [[W7:%.+]] = vector.insert [[F7]], [[W6]]
5656

57-
// CHECK: [[V2:%.+]] = vector.extract_strided_slice [[V]] {offsets = [8], sizes = [1], strides = [1]} : vector<9xf8E4M3FNUZ> to vector<1xf8E4M3FNUZ>
57+
// CHECK: [[V2:%.+]] = vector.extract_strided_slice [[V]][8:1:1] : vector<9xf8E4M3FNUZ> to vector<1xf8E4M3FNUZ>
5858
// CHECK: [[F8:%.+]] = amdgpu.ext_packed_fp8 [[V2]][0]
5959
// CHECK: [[W8:%.+]] = vector.insert [[F8]], [[W7]]
6060
// CHECK: return [[W8]]
@@ -87,7 +87,7 @@ func.func @scalar_trunc(%v: f16) -> f8E5M2FNUZ {
8787
// CHECK: [[V1:%.+]] = vector.extract [[V]][1]
8888
// CHECK: [[F1:%.+]] = arith.truncf [[V1]] : f64 to f32
8989
// CHECK: [[W0:%.+]] = amdgpu.packed_trunc_2xfp8 [[F0]], [[F1]] into undef[word 0] : f32 to vector<4xf8E5M2FNUZ>
90-
// CHECK: [[W:%.+]] = vector.extract_strided_slice [[W0]] {offsets = [0], sizes = [2], strides = [1]} : vector<4xf8E5M2FNUZ> to vector<2xf8E5M2FNUZ>
90+
// CHECK: [[W:%.+]] = vector.extract_strided_slice [[W0]][0:2:1] : vector<4xf8E5M2FNUZ> to vector<2xf8E5M2FNUZ>
9191
// CHECK: return [[W]] : vector<2xf8E5M2FNUZ>
9292
func.func @vector_trunc_short(%v: vector<2xf64>) -> vector<2xf8E5M2FNUZ> {
9393
%w = arith.truncf %v : vector<2xf64> to vector<2xf8E5M2FNUZ>
@@ -101,15 +101,15 @@ func.func @vector_trunc_short(%v: vector<2xf64>) -> vector<2xf8E5M2FNUZ> {
101101
// CHECK: [[ZEROES:%.+]] = arith.constant dense<0.000000e+00> : vector<9xf8E4M3FNUZ>
102102
// CHECK: [[T0:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, %{{.+}} into undef[word 0]
103103
// CHECK: [[T1:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, %{{.+}} into [[T0]][word 1]
104-
// CHECK: [[W0:%.+]] = vector.insert_strided_slice [[T1]], [[ZEROES]] {offsets = [0], strides = [1]}
104+
// CHECK: [[W0:%.+]] = vector.insert_strided_slice [[T1]], [[ZEROES]][0:1]
105105

106106
// CHECK: [[T2:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, %{{.+}} into undef[word 0]
107107
// CHECK: [[T3:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, %{{.+}} into [[T2]][word 1]
108-
// CHECK: [[W1:%.+]] = vector.insert_strided_slice [[T3]], [[W0]] {offsets = [4], strides = [1]}
108+
// CHECK: [[W1:%.+]] = vector.insert_strided_slice [[T3]], [[W0]][4:1]
109109

110110
// CHECK: [[T4:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, undef into undef[word 0]
111-
// CHECK: [[T4_SHORT:%.+]] = vector.extract_strided_slice [[T4]] {offsets = [0], sizes = [1], strides = [1]}
112-
// CHECK: [[W:%.+]] = vector.insert_strided_slice [[T4_SHORT]], [[W1]] {offsets = [8], strides = [1]}
111+
// CHECK: [[T4_SHORT:%.+]] = vector.extract_strided_slice [[T4]][0:1:1]
112+
// CHECK: [[W:%.+]] = vector.insert_strided_slice [[T4_SHORT]], [[W1]][8:1]
113113
// CHECK: return [[W]]
114114
func.func @vector_trunc_long(%v: vector<9xf32>) -> vector<9xf8E4M3FNUZ> {
115115
%w = arith.truncf %v : vector<9xf32> to vector<9xf8E4M3FNUZ>
@@ -123,15 +123,15 @@ func.func @vector_trunc_long(%v: vector<9xf32>) -> vector<9xf8E4M3FNUZ> {
123123
// CHECK: [[ZEROES:%.+]] = arith.constant dense<0.000000e+00> : vector<9xf8E4M3FNUZ>
124124
// CHECK: [[T0:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, %{{.+}} into undef[word 0]
125125
// CHECK: [[T1:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, %{{.+}} into [[T0]][word 1]
126-
// CHECK: [[W0:%.+]] = vector.insert_strided_slice [[T1]], [[ZEROES]] {offsets = [0], strides = [1]}
126+
// CHECK: [[W0:%.+]] = vector.insert_strided_slice [[T1]], [[ZEROES]][0:1]
127127

128128
// CHECK: [[T2:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, %{{.+}} into undef[word 0]
129129
// CHECK: [[T3:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, %{{.+}} into [[T2]][word 1]
130-
// CHECK: [[W1:%.+]] = vector.insert_strided_slice [[T3]], [[W0]] {offsets = [4], strides = [1]}
130+
// CHECK: [[W1:%.+]] = vector.insert_strided_slice [[T3]], [[W0]][4:1]
131131

132132
// CHECK: [[T4:%.+]] = amdgpu.packed_trunc_2xfp8 %{{.+}}, undef into undef[word 0]
133-
// CHECK: [[T4_SHORT:%.+]] = vector.extract_strided_slice [[T4]] {offsets = [0], sizes = [1], strides = [1]}
134-
// CHECK: [[W:%.+]] = vector.insert_strided_slice [[T4_SHORT]], [[W1]] {offsets = [8], strides = [1]}
133+
// CHECK: [[T4_SHORT:%.+]] = vector.extract_strided_slice [[T4]][0:1:1]
134+
// CHECK: [[W:%.+]] = vector.insert_strided_slice [[T4_SHORT]], [[W1]][8:1]
135135
// CHECK: [[RE:%.+]] = vector.shape_cast [[W]] : vector<9xf8E4M3FNUZ> to vector<1x9xf8E4M3FNUZ>
136136
// CHECK: return [[RE]]
137137
func.func @vector_trunc_long_2d(%v: vector<1x9xf32>) -> vector<1x9xf8E4M3FNUZ> {
@@ -144,7 +144,7 @@ func.func @vector_trunc_long_2d(%v: vector<1x9xf32>) -> vector<1x9xf8E4M3FNUZ> {
144144
// CHECK-LABEL: func.func @vector_ext_long_2d
145145
// CHECK-SAME: ([[V:%.+]]: vector<1x9xf8E4M3FNUZ>)
146146
// CHECK: [[CAST:%.+]] = vector.shape_cast [[V]] : vector<1x9xf8E4M3FNUZ> to vector<9xf8E4M3FNUZ>
147-
// CHECK: [[V0:%.+]] = vector.extract_strided_slice [[CAST]] {offsets = [0], sizes = [4], strides = [1]}
147+
// CHECK: [[V0:%.+]] = vector.extract_strided_slice [[CAST]][0:4:1]
148148
// CHECK: [[F0:%.+]] = amdgpu.ext_packed_fp8 [[V0]][0]
149149
// CHECK: [[W0:%.+]] = vector.insert [[F0]]
150150
// CHECK: [[F1:%.+]] = amdgpu.ext_packed_fp8 [[V0]][1]
@@ -154,7 +154,7 @@ func.func @vector_trunc_long_2d(%v: vector<1x9xf32>) -> vector<1x9xf8E4M3FNUZ> {
154154
// CHECK: [[F3:%.+]] = amdgpu.ext_packed_fp8 [[V0]][3]
155155
// CHECK: [[W3:%.+]] = vector.insert [[F3]], [[W2]]
156156

157-
// CHECK: [[V1:%.+]] = vector.extract_strided_slice [[CAST]] {offsets = [4], sizes = [4], strides = [1]} : vector<9xf8E4M3FNUZ> to vector<4xf8E4M3FNUZ>
157+
// CHECK: [[V1:%.+]] = vector.extract_strided_slice [[CAST]][4:4:1] : vector<9xf8E4M3FNUZ> to vector<4xf8E4M3FNUZ>
158158
// CHECK: [[F4:%.+]] = amdgpu.ext_packed_fp8 [[V1]][0]
159159
// CHECK: [[W4:%.+]] = vector.insert [[F4]], [[W3]]
160160
// CHECK: [[F5:%.+]] = amdgpu.ext_packed_fp8 [[V1]][1]
@@ -164,7 +164,7 @@ func.func @vector_trunc_long_2d(%v: vector<1x9xf32>) -> vector<1x9xf8E4M3FNUZ> {
164164
// CHECK: [[F7:%.+]] = amdgpu.ext_packed_fp8 [[V1]][3]
165165
// CHECK: [[W7:%.+]] = vector.insert [[F7]], [[W6]]
166166

167-
// CHECK: [[V2:%.+]] = vector.extract_strided_slice [[CAST]] {offsets = [8], sizes = [1], strides = [1]} : vector<9xf8E4M3FNUZ> to vector<1xf8E4M3FNUZ>
167+
// CHECK: [[V2:%.+]] = vector.extract_strided_slice [[CAST]][8:1:1] : vector<9xf8E4M3FNUZ> to vector<1xf8E4M3FNUZ>
168168
// CHECK: [[F8:%.+]] = amdgpu.ext_packed_fp8 [[V2]][0]
169169
// CHECK: [[W8:%.+]] = vector.insert [[F8]], [[W7]]
170170
// CHECK: [[CAST:%.+]] = vector.shape_cast [[W8]] : vector<9xf32> to vector<1x9xf32>

0 commit comments

Comments
 (0)