Repo sync (#759)

secretflow · Jul 8, 2024 · 54c0885 · 54c0885
1 parent 4ecd05b
commit 54c0885
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/libspu/compiler/tests/interpret/power.mlir b/libspu/compiler/tests/interpret/power.mlir
@@ -58,7 +58,7 @@ func.func @power_op_test_f64_f64_pp() {
    %1 = pphlo.constant dense<[2.0, 2.0, 2.0, -1.0, 1.0]> : tensor<5xf64>
    %2 = pphlo.power %0,%1 : (tensor<5xf64>,tensor<5xf64>)->tensor<5xf64>
    %3 = pphlo.constant dense<[4.000000e+00, 0.000000e+00, 2.500000e+01, 0.33333333333333331, 10000.0]> : tensor<5xf64>
-   pphlo.custom_call @expect_almost_eq(%2, %3) { tol = 0.5 }: (tensor<5xf64>, tensor<5xf64>)->()
+   pphlo.custom_call @expect_almost_eq(%2, %3) { tol = 0.6 }: (tensor<5xf64>, tensor<5xf64>)->()
    func.return
 }
 
@@ -72,6 +72,6 @@ func.func @power_op_test_f64_f64_ss() {
    %4 = pphlo.power %2, %3 : (tensor<5x!pphlo.secret<f64>>,tensor<5x!pphlo.secret<f64>>)->tensor<5x!pphlo.secret<f64>>
    %5 = pphlo.constant dense<[4.000000e+00, 0.000000e+00, 2.500000e+01, 0.33333333333333331, 10000.0]> : tensor<5xf64>
    %6 = pphlo.convert %4 : (tensor<5x!pphlo.secret<f64>>)->tensor<5xf64>
-   pphlo.custom_call @expect_almost_eq(%5, %6) { tol = 0.5 }: (tensor<5xf64>, tensor<5xf64>)->()
+   pphlo.custom_call @expect_almost_eq(%5, %6) { tol = 0.6 }: (tensor<5xf64>, tensor<5xf64>)->()
    func.return
 }
diff --git a/libspu/compiler/tests/interpret/test_json/power.json b/libspu/compiler/tests/interpret/test_json/power.json
@@ -65,7 +65,7 @@
                 }
             ],
             "checker": "expect_almost_eq",
-            "tol": 0.5
+            "tol": 0.6
         }
     ]
 }
diff --git a/spu/tests/BUILD.bazel b/spu/tests/BUILD.bazel
@@ -88,7 +88,6 @@ py_test(
 py_test(
     name = "jnp_cheetah_r64_test",
     size = "enormous",
-    timeout = "long",
     srcs = ["jnp_cheetah_r64_test.py"],
     deps = [
         ":jnp_testbase",