[mlir][sparse] Enable the test for operator expm1.

Reviewed By: aartbik Differential Revision: https://reviews.llvm.org/D126732
llvm · Jun 1, 2022 · 548f084 · 548f084
1 parent c9e242f
commit 548f084
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_complex_ops.mlir b/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_complex_ops.mlir
@@ -99,10 +99,8 @@ module {
         outs(%xv: tensor<?xcomplex<f64>, #SparseVector>) {
         ^bb(%a: complex<f64>, %x: complex<f64>):
           %1 = complex.log1p %a : complex<f64>
-          // TODO(bixia): Enable this line after adding complex.expm1 to
-          // complex to standard lowering.
-          // %2 = complex.expm1 %1 : complex<f64>
-          linalg.yield %1 : complex<f64>
+          %2 = complex.expm1 %1 : complex<f64>
+          linalg.yield %2 : complex<f64>
     } -> tensor<?xcomplex<f64>, #SparseVector>
     return %0 : tensor<?xcomplex<f64>, #SparseVector>
   }
@@ -225,12 +223,12 @@ module {
     // CHECK-NEXT: 0.995055
     // CHECK-NEXT: 0
     call @dumpc(%3, %d3) : (tensor<?xcomplex<f64>, #SparseVector>, index) -> ()
-    // CHECK-NEXT: 1.52361
-    // CHECK-NEXT: 2.69061
-    // CHECK-NEXT: 1.73287
-    // CHECK-NEXT: 0.785398
-    // CHECK-NEXT: 2.13833
-    // CHECK-NEXT: 0.785398
+    // CHECK-NEXT: -5.13
+    // CHECK-NEXT: 2
+    // CHECK-NEXT: 3
+    // CHECK-NEXT: 4
+    // CHECK-NEXT: 5
+    // CHECK-NEXT: 6
     call @dumpc(%4, %d3) : (tensor<?xcomplex<f64>, #SparseVector>, index) -> ()
     // CHECK-NEXT: -2.565
     // CHECK-NEXT: 1