Merge branch 'master' of github.com:LatticeX-Foundation/Rosetta

LatticeX-Foundation · Jul 30, 2021 · ccbf79a · ccbf79a
2 parents f90acf1 + d698a86
commit ccbf79a
Show file tree

Hide file tree

Showing 11 changed files with 23 additions and 13 deletions.
diff --git a/python/latticex/rosetta/test/spass/test_di.py b/python/latticex/rosetta/test/spass/test_di.py
@@ -52,5 +52,7 @@
         print("Fail")
 
 
+rtt.deactivate()
+
 Writer = tf.summary.FileWriter("log/di", tf.get_default_graph())
 Writer.close()
diff --git a/python/latticex/rosetta/test/spass/test_graph_opt.py b/python/latticex/rosetta/test/spass/test_graph_opt.py
@@ -10,7 +10,7 @@
 
 
 # rtt.set_backend_loglevel(0)
-
+rtt.activate("SecureNN")
 X = tf.Variable([[2., 3.], [4., 5.]], tf.float64, name="x")
 Y = tf.Variable([[1.], [1.]], tf.float64, name="y")
 W = tf.Variable(tf.zeros([2, 1], dtype=tf.float64), name="w")
@@ -39,7 +39,7 @@
     print("Fail")
 
 
-
+rtt.deactivate()
 
 Writer = tf.summary.FileWriter("log/graph_opt", tf.get_default_graph())
 Writer.close()
diff --git a/python/latticex/rosetta/test/spass/test_graph_opt2.py b/python/latticex/rosetta/test/spass/test_graph_opt2.py
@@ -10,7 +10,7 @@
 
 
 # rtt.set_backend_loglevel(0)
-
+rtt.activate("SecureNN")
 X = tf.Variable([[2., 3.], [4., 5.]], tf.float64, name="x")
 Y = tf.Variable([[1.], [1.]], tf.float64, name="y")
 W = tf.Variable(tf.zeros([2, 1], dtype=tf.float64), name="w")
@@ -41,7 +41,7 @@
     print("Fail")
 
 
-
+rtt.deactivate()
 
 Writer = tf.summary.FileWriter("log/graph_opt2", tf.get_default_graph())
 Writer.close()
diff --git a/python/latticex/rosetta/test/spass/test_graph_opt3.py b/python/latticex/rosetta/test/spass/test_graph_opt3.py
@@ -81,7 +81,7 @@
     print("Fail")
 
 
-
+rtt.deactivate()
 
 Writer = tf.summary.FileWriter("log/graph_opt3", tf.get_default_graph())
 Writer.close()

diff --git a/python/latticex/rosetta/test/spass/test_iter.py b/python/latticex/rosetta/test/spass/test_iter.py
@@ -53,6 +53,6 @@
     print("Fail")
 
 
-
+rtt.deactivate()
 Writer = tf.summary.FileWriter("log/iter", tf.get_default_graph())
 Writer.close()
diff --git a/python/latticex/rosetta/test/spass/test_iter2.py b/python/latticex/rosetta/test/spass/test_iter2.py
@@ -7,6 +7,8 @@
 with_index = False
 batch_size = 2
 
+rtt.activate("SecureNN")
+
 def decode(line):
     fields = tf.string_split([line], ',').values
     if with_index: # Skip index
@@ -54,5 +56,8 @@ def decode(line):
 except:
     print("Fail")
 
+
+rtt.deactivate()
+
 Writer = tf.compat.v1.summary.FileWriter("log/iter2", tf.compat.v1.get_default_graph())
 Writer.close()
diff --git a/python/latticex/rosetta/test/spass/test_mpc_feed.py b/python/latticex/rosetta/test/spass/test_mpc_feed.py
@@ -3,7 +3,7 @@
 import numpy as np
 
 
-
+rst.activate("SecureNN")
 X = tf.placeholder(tf.float64, [2])
 print(X)
 Y = tf.Variable(tf.ones([2], dtype=tf.float64))
@@ -29,6 +29,6 @@
     print("Fail")
 
 
-
+rst.deactivate()
 Writer = tf.summary.FileWriter("log/feed", tf.get_default_graph())
 Writer.close()
diff --git a/python/latticex/rosetta/test/spass/test_mpc_feed2.py b/python/latticex/rosetta/test/spass/test_mpc_feed2.py
@@ -3,7 +3,7 @@
 import numpy as np
 
 
-
+rst.activate("SecureNN")
 X = tf.placeholder(tf.float64, [2, 2])
 Y = tf.placeholder(tf.float64, [2, 1])
 
@@ -43,6 +43,6 @@
     print("Fail")
 
 
-
+rst.deactivate()
 Writer = tf.summary.FileWriter("log/feed2", tf.get_default_graph())
 Writer.close()
diff --git a/python/latticex/rosetta/test/spass/test_mpc_predict.py b/python/latticex/rosetta/test/spass/test_mpc_predict.py
@@ -2,6 +2,7 @@
 import latticex.rosetta as rst
 
 
+rst.activate()
 X = tf.Variable(1.0, name='x')
 Y = tf.Variable(2.0, name='y')
 Z = tf.multiply(X, Y)
@@ -19,7 +20,7 @@
     print("Fail")
 
 
-
+rst.deactivate()
 Writer = tf.summary.FileWriter("log/pridict", tf.get_default_graph())
 Writer.close()
 

diff --git a/python/latticex/rosetta/test/spass/test_mpc_reveal.py b/python/latticex/rosetta/test/spass/test_mpc_reveal.py
@@ -2,6 +2,7 @@
 import latticex.rosetta as rst
 
 
+rst.activate("SecureNN")
 xa = tf.Variable(
     [
         [1.892, 2],
@@ -35,6 +36,6 @@
     print("Fail")
 
 
-
+rst.deactivate()
 Writer = tf.summary.FileWriter("log/reveal", tf.get_default_graph())
 Writer.close()
diff --git a/python/latticex/rosetta/test/spass/test_node_dup.py b/python/latticex/rosetta/test/spass/test_node_dup.py
@@ -13,7 +13,7 @@
 import numpy as np
 np.set_printoptions(suppress=True)
 
-
+rtt.activate("SecureNN")
 xa = tf.Variable(
     [
         [1.892, 2],
@@ -58,5 +58,6 @@
     print("Fail")
 
 
+rtt.deactivate()
 Writer = tf.summary.FileWriter("log/node_dup", tf.get_default_graph())
 Writer.close()