Merge pull request #239 from kaih70/master

fix link and ut bugs
PaddlePaddle · Nov 30, 2021 · 995faa8 · 995faa8
2 parents a6f8664 + 395d2df
commit 995faa8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/core/paddlefl_mpc/mpc_protocol/CMakeLists.txt b/core/paddlefl_mpc/mpc_protocol/CMakeLists.txt
@@ -46,17 +46,18 @@ if (WITH_GPU)
     cuda_add_library(mpc_protocol_cu SHARED ${PROTO_SRCS_CU})
 endif()
 
+if (WITH_GRPC)
+    target_link_libraries(mpc_protocol transport_o grpc++)
+    cc_test(mesh_network_grpc_test SRCS network/mesh_network_grpc_test.cc DEPS mpc_protocol)
+endif()
+
 if (WITH_GPU)
     target_link_libraries(mpc_protocol ${GRPC_DEPS} gloo hiredis privc3 fluid_framework mpc_protocol_cu)
     target_link_libraries(mpc_protocol_cu ${GRPC_DEPS} gloo hiredis privc3 common_cu)
 else ()
     target_link_libraries(mpc_protocol ${GRPC_DEPS} gloo hiredis privc3 privc_o fluid_framework)
 endif()
 
-if (WITH_GRPC)
-    target_link_libraries(mpc_protocol transport_o)
-    cc_test(mesh_network_grpc_test SRCS network/mesh_network_grpc_test.cc DEPS mpc_protocol)
-endif()
 cc_test(mesh_network_test SRCS network/mesh_network_test.cc DEPS mpc_protocol)
 if (NOT WITH_GPU)
     cc_test(mpc_protocol_test SRCS mpc_protocol_test.cc DEPS mpc_protocol privc)

diff --git a/python/paddle_fl/mpc/tests/unittests/test_op_batch_norm.py b/python/paddle_fl/mpc/tests/unittests/test_op_batch_norm.py
@@ -57,7 +57,7 @@ def batch_norm(self, **kwargs):
     def test_batch_norm(self):
 
         data_1 = np.array(
-            [[10, 10, 10], [50, 50, 50]]).astype('float32')
+            [[1, 1, 1], [5, 5, 5]]).astype('float32')
 
         expected_out = np.array(
             [[-1, -1, -1], [1, 1, 1]]).astype('float32')