33 lines
608 B
Python
33 lines
608 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import multiprocessing
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
def doprocess(id):
|
||
|
|
|
||
|
|
import time
|
||
|
|
|
||
|
|
sys.argv.extend(["--node_id", "P{}".format(id)])
|
||
|
|
import latticex.rosetta as rtt
|
||
|
|
|
||
|
|
import tensorflow as tf
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
rtt.activate("SecureNN")
|
||
|
|
|
||
|
|
p0 = multiprocessing.Process(target=doprocess, args=(0,))
|
||
|
|
p1 = multiprocessing.Process(target=doprocess, args=(1,))
|
||
|
|
p2 = multiprocessing.Process(target=doprocess, args=(2,))
|
||
|
|
|
||
|
|
p0.daemon = True
|
||
|
|
p0.start()
|
||
|
|
p1.daemon = True
|
||
|
|
p1.start()
|
||
|
|
p2.daemon = True
|
||
|
|
p2.start()
|
||
|
|
|
||
|
|
|
||
|
|
p0.join()
|
||
|
|
p1.join()
|
||
|
|
p2.join()
|