Support QDQ decomposition of TanOp

openxla · Jul 26, 2024 · 90bb15a · 90bb15a
1 parent e080c8f
commit 90bb15a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/stablehlo/tests/stablehlo_legalize_quant_to_int.mlir b/stablehlo/tests/stablehlo_legalize_quant_to_int.mlir
@@ -2388,12 +2388,16 @@ func.func @send(%arg0: tensor<1x2x2x!quant.uniform<i8:f32, 1.0:17>>, %token0: !s
 }
 
 // -----
+
+// CHECK-LABEL:  func.func @tan
 func.func @tan(%arg0: tensor<1x2x2x!quant.uniform<i8:f32, 1.0:17>>) -> tensor<1x2x2x!quant.uniform<i8:f32, 1.0:17>> {
-  // expected-error@+1 {{failed to legalize operation 'stablehlo.tan' that was explicitly marked illegal}}
   %0 = "stablehlo.tan"(%arg0) : (tensor<1x2x2x!quant.uniform<i8:f32, 1.0:17>>) -> tensor<1x2x2x!quant.uniform<i8:f32, 1.0:17>>
   func.return %0 : tensor<1x2x2x!quant.uniform<i8:f32, 1.0:17>>
 }
 
+// CHECK: stablehlo.tan {{.*}} : tensor<1x2x2xf32>
+
+
 // -----
 func.func @tuple(%arg0: tensor<1x2x2x!quant.uniform<i8:f32, 1.0:17>>,
                  %arg1: tensor<1x2x2x!quant.uniform<i8:f32, 1.0:17>>) {