@@ -3236,23 +3236,19 @@ func.func @rank_1_shuffle_to_interleave(%arg0: vector<6xi32>, %arg1: vector<6xi3
3236
3236
3237
3237
// -----
3238
3238
3239
- // CHECK-LABEL: func @extract_from_0d_splatlike_broadcast_regression (
3239
+ // CHECK-LABEL: func @extract_from_splatlike_broadcast (
3240
3240
// CHECK-SAME: %[[A:.*]]: f32, %[[B:.*]]: vector<f32>, %[[C:.*]]: vector<2xf32>)
3241
- func.func @extract_from_0d_splatlike_broadcast_regression (%a: f32 , %b: vector <f32 >, %c: vector <2 xf32 >) -> (f32 , f32 , f32 , f32 , f32 , vector <6 x7 xf32 >, vector <3 xf32 >) {
3242
- // Splat/broadcast scalar to 0D and extract scalar.
3241
+ func.func @extract_from_splatlike_broadcast (%a: f32 , %b: vector <f32 >, %c: vector <2 xf32 >) -> (f32 , f32 , f32 , f32 , vector <6 x7 xf32 >, vector <3 xf32 >) {
3242
+ // Broadcast scalar to 0D and extract scalar.
3243
3243
%0 = vector.broadcast %a : f32 to vector <f32 >
3244
3244
%1 = vector.extract %0 [] : f32 from vector <f32 >
3245
3245
3246
- // Broadcast scalar to 0D and extract scalar.
3247
- %2 = vector.broadcast %a : f32 to vector <f32 >
3248
- %3 = vector.extract %2 [] : f32 from vector <f32 >
3249
-
3250
3246
// Broadcast 0D to 3D and extract scalar.
3251
3247
// CHECK: %[[EXTRACT1:.*]] = vector.extract %[[B]][] : f32 from vector<f32>
3252
3248
%4 = vector.broadcast %b : vector <f32 > to vector <1 x2 x4 xf32 >
3253
3249
%5 = vector.extract %4 [0 , 0 , 1 ] : f32 from vector <1 x2 x4 xf32 >
3254
3250
3255
- // Splat/broadcast scalar to 2D and extract scalar.
3251
+ // Broadcast scalar to 2D and extract scalar.
3256
3252
%6 = vector.broadcast %a : f32 to vector <2 x3 xf32 >
3257
3253
%7 = vector.extract %6 [0 , 1 ] : f32 from vector <2 x3 xf32 >
3258
3254
@@ -3268,8 +3264,8 @@ func.func @extract_from_0d_splatlike_broadcast_regression(%a: f32, %b: vector<f3
3268
3264
// CHECK: %[[EXTRACT3:.*]] = vector.broadcast %[[A]] : f32 to vector<3xf32>
3269
3265
%11 = vector.extract %6 [1 ] : vector <3 xf32 > from vector <2 x3 xf32 >
3270
3266
3271
- // CHECK: return %[[A]], %[[A]], %[[ EXTRACT1]], %[[A]], %[[A]], %[[EXTRACT2]], %[[EXTRACT3]]
3272
- return %1 , %3 , % 5 , %7 , %9 , %10 , %11 : f32 , f32 , f32 , f32 , f32 , vector <6 x7 xf32 >, vector <3 xf32 >
3267
+ // CHECK: return %[[A]], %[[EXTRACT1]], %[[A]], %[[A]], %[[EXTRACT2]], %[[EXTRACT3]]
3268
+ return %1 , %5 , %7 , %9 , %10 , %11 : f32 , f32 , f32 , f32 , vector <6 x7 xf32 >, vector <3 xf32 >
3273
3269
}
3274
3270
3275
3271
// -----
0 commit comments