repo-sync-2024-07-08T14:01:52+0800

anakinxc · Jul 8, 2024 · 9f35708 · 9f35708
1 parent db7b053
commit 9f35708
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/libspu/mpc/cheetah/boolean_semi2k.cc b/libspu/mpc/cheetah/boolean_semi2k.cc
@@ -81,8 +81,8 @@ NdArrayRef P2B::proc(KernelEvalContext* ctx, const NdArrayRef& in) const {
   if (comm->getRank() == 0) {
     ring_xor_(x, in);
   }
-
-  return makeBShare(x, field, getNumBits(in));
+  auto nbits = getNumBits(in) == 0 ? 1 : getNumBits(in);
+  return makeBShare(x, field, nbits);
 }
 
 NdArrayRef AndBP::proc(KernelEvalContext* ctx, const NdArrayRef& lhs,

diff --git a/spu/tests/BUILD.bazel b/spu/tests/BUILD.bazel
@@ -87,6 +87,7 @@ py_test(
 
 py_test(
     name = "jnp_cheetah_r64_test",
+    size = "enormous",
     timeout = "long",
     srcs = ["jnp_cheetah_r64_test.py"],
     deps = [
@@ -96,7 +97,7 @@ py_test(
 
 py_test(
     name = "jnp_cheetah_r64_test_x64",
-    timeout = "long",
+    size = "enormous",
     srcs = ["jnp_cheetah_r64_test.py"],
     env = {
         "ENABLE_X64_TEST": "1",

diff --git a/spu/tests/jnp_cheetah_r64_test.py b/spu/tests/jnp_cheetah_r64_test.py
@@ -22,8 +22,7 @@
 from spu.tests.jnp_testbase import JnpTests
 
 
-@unittest.skip("too slow, last run succeed")
-class JnpTestAby3FM64(JnpTests.JnpTestBase):
+class JnpTestCheetahFM64(JnpTests.JnpTestBase):
     def setUp(self):
         self._sim = ppsim.Simulator.simple(
             2, spu_pb2.ProtocolKind.CHEETAH, spu_pb2.FieldType.FM64