mirror of
https://github.com/data61/MP-SPDZ.git
synced 2026-01-10 14:08:09 -05:00
63 lines
1.6 KiB
Python
Executable File
63 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys, os
|
|
import collections
|
|
|
|
sys.path.append('.')
|
|
|
|
from Compiler.program import *
|
|
from Compiler.instructions_base import *
|
|
|
|
if len(sys.argv) <= 1:
|
|
print('Usage: %s <program>' % sys.argv[0])
|
|
|
|
res = collections.defaultdict(lambda: 0)
|
|
regs = collections.defaultdict(lambda: 0)
|
|
thread_regs = collections.defaultdict(lambda: 0)
|
|
|
|
def process(tapename, res, regs):
|
|
for inst in Tape.read_instructions(tapename):
|
|
t = inst.type
|
|
if issubclass(t, DirectMemoryInstruction):
|
|
res[type(inst.args[0])] = max(inst.args[1].i + inst.size,
|
|
res[type(inst.args[0])]) + 1
|
|
for arg in inst.args:
|
|
if isinstance(arg, RegisterArgFormat):
|
|
regs[type(arg)] = max(regs[type(arg)], arg.i + inst.size)
|
|
|
|
tapes = Program.read_tapes(sys.argv[1])
|
|
|
|
process(next(tapes), res, regs)
|
|
|
|
for tapename in tapes:
|
|
process(tapename, res, thread_regs)
|
|
|
|
reverse_formats = dict((v, k) for k, v in ArgFormats.items())
|
|
|
|
regout = lambda regs: dict((reverse_formats[t], n) for t, n in regs.items())
|
|
|
|
def output(data):
|
|
for t, n in data.items():
|
|
if n:
|
|
try:
|
|
print('%10d %s' % (n, ArgFormats[t.removesuffix('w')].name))
|
|
except:
|
|
pass
|
|
|
|
total = 0
|
|
for x in res, regs, thread_regs:
|
|
total += sum(x.values())
|
|
|
|
print ('Memory:')
|
|
output(regout(res))
|
|
|
|
print ('Registers in main thread:')
|
|
output(regout(regs))
|
|
|
|
if thread_regs:
|
|
print ('Registers in other threads:')
|
|
output(regout(thread_regs))
|
|
|
|
print ('The program requires at the very least %f GB of RAM per party.' % \
|
|
(total * 8e-9))
|