Fix bugs in script for generating configuration files

ucbrise · May 23, 2021 · c40ebd9 · c40ebd9
1 parent 75dc4cd
commit c40ebd9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/generate_configs.py b/scripts/generate_configs.py
@@ -165,20 +165,20 @@ def generate_paired_wan_config_dict(protocol, scenario, num_workers_per_party, i
                                 size = "30gb"
                         elif scenario == "unbounded":
                             size = "4096gb" # Larger than the standard "unbounded" size
-                        config_dict = generate_paired_wan_config_dict(size, party_size, id, azure_ids, gcloud_ids, cluster, *ot_params)
+                        config_dict = generate_paired_wan_config_dict("halfgates", size, party_size, id, azure_ids, gcloud_ids, cluster, *ot_params)
                         output_path = os.path.join(output_dir_path, "config_halfgates_{0}_{1}_{2}.yaml".format(party_size, ot_pipeline_depth, ot_num_daemons))
                         with open(output_path, "w") as f:
                             yaml.dump(config_dict, f, sort_keys = False, default_flow_style = False)
     else:
         azure_id = id
         gcloud_id = cluster["location_to_id"][location]
-        for protocol in ("halfgates", "ckks"):
+        for protocol in ("halfgates",):
             for scenario, output_dir_path in zip(memory_bounds, dir_paths):
                 for party_size in (1, 2, 4, 8, 16):
                     for ot_pipeline_depth in tuple(2 ** i for i in range(9)):
                         for ot_num_daemons in tuple(2 ** i for i in range(9)):
                             ot_params = (ot_pipeline_depth, ot_num_daemons)
-                            config_dict = generate_wan_config_dict("halfgates", scenario, party_size, azure_id, gcloud_id, cluster, *ot_params)
+                            config_dict = generate_wan_config_dict(protocol, scenario, party_size, azure_id, gcloud_id, cluster, *ot_params)
                             output_path = os.path.join(output_dir_path, "config_halfgates_{0}_{1}_{2}.yaml".format(party_size, ot_pipeline_depth, ot_num_daemons))
                             with open(output_path, "w") as f:
                                 yaml.dump(config_dict, f, sort_keys = False, default_flow_style = False)