debug: benchmark

lijin456 · Oct 6, 2021 · 14dd6e3 · 14dd6e3
1 parent c674135
commit 14dd6e3
Show file tree

Hide file tree

Showing 4 changed files with 53 additions and 11 deletions.
diff --git a/files/parties_WAN_3PC b/files/parties_WAN_3PC
@@ -1,3 +1,3 @@
-35.166.8.86
-18.218.6.145
-54.153.35.13
+172.31.163.52
+172.31.163.53
+172.31.163.54
diff --git a/src/Functionalities.cpp b/src/Functionalities.cpp
@@ -1664,7 +1664,8 @@ void debugDotProd()
 	// 	for (size_t i = 0; i < size; ++i)
 	// 		b[i] = aes_indep->get64Bits();
 
-	funcDotProductMPC(a, b, c, size);
+    for(int i = 0; i < 1000 ;i++)
+        funcDotProductMPC(a, b, c, size);
 
 	if (PRIMARY)
 		funcReconstruct2PC(c, size, "c");
@@ -1847,7 +1848,8 @@ void debugSS()
 			for (size_t i = 0; i < size; ++i)
 				inputs[i] = (myType)aes_indep->get8Bits();
 
-		funcSelectShares3PC(inputs, selector, outputs, size);
+        for(int i = 0; i < 1000 ;i++)
+            funcSelectShares3PC(inputs, selector, outputs, size);
 
 		if (PRIMARY)
 		{
@@ -1894,6 +1896,39 @@ void debugMatMul()
 		funcReconstruct2PC(c, c.size(), "c");
 }
 
+void debugReLU()
+{
+    size_t size = 10;
+    vector<myType> inputs(size, 0);
+
+    if (partyNum == PARTY_A)
+        for (size_t i = 0; i < size; ++i)
+            inputs[i] = aes_indep->get8Bits() - aes_indep->get8Bits();
+
+    if (THREE_PC)
+    {
+        vector<myType> outputs(size, 0);
+        for (int i = 0; i< 1000;i++)
+            funcRELUMPC(inputs, outputs, size);
+        if (PRIMARY)
+        {
+            funcReconstruct2PC(inputs, size, "inputs");
+            funcReconstruct2PC(outputs, size, "outputs");
+        }
+    }
+
+    if (FOUR_PC)
+    {
+        vector<smallType> outputs(size, 0);
+        funcRELUPrime4PC(inputs, outputs, size);
+        if (PRIMARY)
+        {
+            funcReconstruct2PC(inputs, size, "inputs");
+            funcReconstructBit2PC(outputs, size, "outputs");
+        }
+    }
+}
+
 void debugReLUPrime()
 {
 	size_t size = 10;
@@ -1906,7 +1941,8 @@ void debugReLUPrime()
 	if (THREE_PC)
 	{
 		vector<myType> outputs(size, 0);
-		funcRELUPrime3PC(inputs, outputs, size);
+		for (int i = 0; i< 1000;i++)
+            funcRELUPrime3PC(inputs, outputs, size);
 		if (PRIMARY)
 		{
 			funcReconstruct2PC(inputs, size, "inputs");

diff --git a/src/Functionalities.h b/src/Functionalities.h
@@ -55,6 +55,7 @@ void debugSS();
 void debugMatMul();
 void debugReLUPrime();
 void debugMaxIndex();
+void debugReLU();
 
 //Test
 void testMatMul(size_t rows, size_t common_dim, size_t columns, size_t iter);

diff --git a/src/main.cpp b/src/main.cpp
@@ -102,9 +102,14 @@ int main(int argc, char** argv)
 		aes_parallel->precompute();
 
 
-/****************************** RUN NETWORK/BENCHMARKS ******************************/ 
-	start_m();
-
+/****************************** RUN NETWORK/BENCHMARKS ******************************/
+    if (!STANDALONE)
+        initializeMPC();
+    start_m();
+//    debugDotProd();
+//    debugSS();
+//    debugReLU();
+    debugReLUPrime();
 	// whichNetwork = "Mat-Mul";
 	// testMatMul(784, 128, 10, NUM_ITERATIONS);
 	// testMatMul(1, 500, 100, NUM_ITERATIONS);
@@ -135,8 +140,8 @@ int main(int argc, char** argv)
 	// testMaxPoolDerivative(24, 24, 2, 2, 16, NUM_ITERATIONS);
 	// testMaxPoolDerivative(8, 8, 4, 4, 50, NUM_ITERATIONS);
 
-	whichNetwork += " train";
-	train(network, config);
+//	whichNetwork += " train";
+//	train(network, config);
 
 	// whichNetwork += " test";
 	// test(network);