[mlir][sparse] Enable the test for operator expm1.
authorbixia1 <bixia@google.com>
Tue, 31 May 2022 23:17:45 +0000 (16:17 -0700)
committerbixia1 <bixia@google.com>
Wed, 1 Jun 2022 18:18:17 +0000 (11:18 -0700)
Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D126732

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_complex_ops.mlir

index 0fbb2b7..fa3d4c6 100644 (file)
@@ -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