From d0efc71796b3f254fc8d79f1c20e37aa3d0ba691 Mon Sep 17 00:00:00 2001 From: Ben Date: Mon, 11 Apr 2022 15:11:03 +0100 Subject: [PATCH] allow for a single arg for AWS --- new_scripts.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/new_scripts.py b/new_scripts.py index 7e3855e61..4232c5066 100644 --- a/new_scripts.py +++ b/new_scripts.py @@ -2,6 +2,7 @@ from estimator_new import * from sage.all import oo, save from math import log2 + def old_models(security_level, sd, logq = 32): """ Use the old model as a starting point for the data gathering step @@ -142,7 +143,7 @@ def automated_param_select_n(params, target_security=128): return params -def generate_parameter_matrix(params_in, sd_range, target_security_levels=[128]): +def generate_parameter_matrix(params_in, sd_range, target_security_levels=[128], name="v0.sobj"): """ :param sd_range: a tuple (sd_min, sd_max) giving the values of sd for which to generate parameters :param params: the standard deviation of the LWE error @@ -164,16 +165,26 @@ def generate_parameter_matrix(params_in, sd_range, target_security_levels=[128]) Xe_new = nd.NoiseDistribution.DiscreteGaussian(2**sd) params_out = automated_param_select_n(params_in.updated(Xe=Xe_new), target_security=lam) results["{}".format(lam)].append((params_out.n, params_out.q, params_out.Xe.stddev)) + save(results, "{}.sobj".format(name)) return results -def generate_zama_curves64(sd_range=[2, 60], target_security_levels=[128, 192, 256]): +def generate_zama_curves64(sd_range=[2, 60], target_security_levels=[256], name="v0256.sobj"): D = ND.DiscreteGaussian init_params = LWE.Parameters(n=1024, q=2 ** 64, Xs=D(0.50, -0.50), Xe=D(131072.00), m=oo, tag='TFHE_DEFAULT') - raw_data = generate_parameter_matrix(init_params, sd_range=sd_range, target_security_levels=target_security_levels) + raw_data = generate_parameter_matrix(init_params, sd_range=sd_range, target_security_levels=target_security_levels, name=name) return raw_data -generate_zama_curves64() \ No newline at end of file +def plota_curve(raw_data, security_level): + + data = raw_data["{}".format(security_level)] + +import sys +a = int(sys.argv[1]) +print(a) + +generate_zama_curves64(target_security_levels=[a]) +