github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-09 22:57:59 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(ci): handle multi directories for parameters records

Browse Source
This commit is contained in:
David Testé
2025-05-22 11:20:03 +02:00
committed by David Testé
parent c19cd9f021
commit 2749c1088c
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
ci/benchmark_parser.py
View File

@@ -14,6 +14,8 @@ import sys
ONE_HOUR_IN_SECONDS = 3600
ONE_SECOND_IN_NANOSECONDS = 1e9
# These are directories where crypto parameters records can be stored.
BENCHMARK_DIRS = ["tfhe-benchmark", "tfhe-zk-pok"]
parser = argparse.ArgumentParser()
parser.add_argument(
@@ -348,8 +350,18 @@ def get_parameters(bench_id, directory):
:return: :class:`tuple` as ``(benchmark parameters, display name, operator type)``
"""
params_dir = pathlib.Path("tfhe-benchmark", "benchmarks_parameters", bench_id)
params = _parse_file_to_json(params_dir, "parameters.json")
for dirname in BENCHMARK_DIRS:
params_dir = pathlib.Path(dirname, "benchmarks_parameters", bench_id)
try:
params = _parse_file_to_json(params_dir, "parameters.json")
except FileNotFoundError:
continue
else:
break
else:
raise FileNotFoundError(
f"file not found: '[...]/benchmarks_parameters/{bench_id}/parameters.json'"
)
display_name = params.pop("display_name")
operator = params.pop("operator_type")