Ryan Rueger

ryan@rueg.re / picture / key / home
aboutsummaryrefslogtreecommitdiffhomepage
path: root/time_pegasis.py
diff options
context:
space:
mode:
Diffstat (limited to 'time_pegasis.py')
-rw-r--r--time_pegasis.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/time_pegasis.py b/time_pegasis.py
index c81c745..3ce605d 100644
--- a/time_pegasis.py
+++ b/time_pegasis.py
@@ -27,12 +27,12 @@ if __name__ == "__main__":
# Arguments
parser = argparse.ArgumentParser()
- parser.add_argument("-p", type=int, help="Prime size", choices=(500, 1000, 1500, 2000, 4000), default=500)
+ parser.add_argument("-p", type=int, help="Prime size", choices=(100, 500, 1000, 1500, 2000, 4000), default=500)
parser.add_argument("--nruns", type=int, help="Number of runs", default=5)
parser.add_argument("--runname", type=str, help="Run name", default="run")
parser.add_argument("--precompute", action=argparse.BooleanOptionalAction)
args = parser.parse_args()
- assert args.p in [500, 1000, 1500, 2000, 4000]
+ assert args.p in [100, 500, 1000, 1500, 2000, 4000]
EGA = PEGASIS(args.p)
if args.precompute: