Skip to content

Commit

Permalink
Merge branch 'tina.FXML-3548-bump-onnx-mlir-to-c03bc49b430014c6d1ae85…
Browse files Browse the repository at this point in the history
…0de50295c99974cc57' into tina.FXML-3548-bump-update-tosa-format-trial-branch
  • Loading branch information
TinaAMD committed Jan 29, 2024
2 parents 7f64498 + cbf0c71 commit d5df402
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions test/mlir/conversion/onnx_to_tosa/Math/Elementwise.mlir
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
// -----

func.func @test_cast_f32_i8(%arg0: tensor<13x21x1xf32>) -> tensor<13x21x1xi8> {
%0 = "onnx.Cast"(%arg0) {to = i8} : (tensor<13x21x1xf32>) -> tensor<13x21x1xi8>
%0 = "onnx.Cast"(%arg0) {to = i8} : (tensor<13x21x1xf32>) -> tensor<13x21x1xi8>
"func.return"(%0) : (tensor<13x21x1xi8>) -> ()
// CHECK-LABEL: func @test_cast_f32_i8
// CHECK-SAME: ([[PARAM_0_:%.+]]: tensor<13x21x1xf32>) -> tensor<13x21x1xi8> {
Expand All @@ -13,7 +13,7 @@ func.func @test_cast_f32_i8(%arg0: tensor<13x21x1xf32>) -> tensor<13x21x1xi8> {
// -----

func.func @test_cast_i8_i1(%arg0: tensor<1x21x1x1xi8>) -> tensor<1x21x1x1xi1> {
%0 = "onnx.Cast"(%arg0) {to = i1} : (tensor<1x21x1x1xi8>) -> tensor<1x21x1x1xi1>
%0 = "onnx.Cast"(%arg0) {to = i1} : (tensor<1x21x1x1xi8>) -> tensor<1x21x1x1xi1>
"func.return"(%0) : (tensor<1x21x1x1xi1>) -> ()
// CHECK-LABEL: func @test_cast_i8_i1
// CHECK-SAME: ([[PARAM_0_:%.+]]: tensor<1x21x1x1xi8>) -> tensor<1x21x1x1xi1> {
Expand Down Expand Up @@ -300,6 +300,7 @@ func.func @test_sigmoid(%arg0 : tensor<10x10xf32>) -> tensor<10x10xf32> {
// CHECK-NEXT: [[VAR_0_:%.+]] = tosa.sigmoid [[PARAM_0_]] : (tensor<10x10xf32>) -> tensor<10x10xf32>
// CHECK-NEXT: return [[VAR_0_]] : tensor<10x10xf32>
// CHECK-NEXT: }
}

// -----

Expand Down Expand Up @@ -941,4 +942,4 @@ func.func @test_less_broadcast(%arg0: tensor<13x21x1xf32>, %arg1: tensor<1xf32>)
// CHECK: [[VAR_0_:%.+]] = tosa.reshape %arg1 {new_shape = array<i64: 1, 1, 1>} : (tensor<1xf32>) -> tensor<1x1x1xf32>
// CHECK: [[VAR_1_:%.+]] = tosa.greater [[VAR_0_]], %arg0 : (tensor<1x1x1xf32>, tensor<13x21x1xf32>) -> tensor<13x21x1xi1>
// CHECK: return [[VAR_1_]] : tensor<13x21x1xi1>
}
}

0 comments on commit d5df402

Please sign in to comment.