Skip to content

Commit 4bd36d5

Browse files
author
Ferdinand Lemaire
committed
Fix reference reshape test
1 parent 9c4184b commit 4bd36d5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/test/Dialect/Tosa/constant-reshape-fold.mlir

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@ func.func @reshape_multi_user_non_splat() -> (tensor<1x2xf32>, tensor<2xf32>) {
2929
// CHECK: %[[CONST:.*]] = "tosa.const"{{.*}}-> tensor<2xf32>
3030
// CHECK: %[[RES:.*]] = tosa.reshape
3131
// CHECK: return %[[RES]], %[[CONST]]
32-
// CHECK-ALWAYS-DAG: %[[RES:.*]] = "tosa.const"{{.*}}-> tensor<2xf32>
33-
// CHECK-ALWAYS-DAG: %[[RESHAPED:.*]] = "tosa.const"{{.*}}-> tensor<1x2xf32>
34-
// CHECK-ALWAYS: return %[[RESHAPED]], %[[RES]]
32+
// CHECK-ALWAYS: %[[CONST_1:.*]] = "tosa.const"() <{value = dense<4.000000e+00> : tensor<2xf32>}> : () -> tensor<2xf32>
33+
// CHECK-ALWAYS: %[[CONST_2:.*]] = "tosa.const"() <{value = dense<4.000000e+00> : tensor<1x2xf32>}> : () -> tensor<1x2xf32>
34+
// CHECK-ALWAYS: return %[[CONST_2]], %[[CONST_1]]
3535
%0 = "tosa.const"() {value = dense<[4.0, 3.0]> : tensor<2xf32>} : () -> tensor<2xf32>
3636
%1 = tosa.reshape %0 {new_shape = array<i64: 1, 2>}: (tensor<2xf32>) -> tensor<1x2xf32>
3737
return %1, %0 : tensor<1x2xf32>, tensor<2xf32>

0 commit comments

Comments
 (0)