mirror of
https://github.com/zama-ai/tfhe-rs.git
synced 2026-01-08 22:28:01 -05:00
chore(ci): provide hardware name for benchmarks with avx512
This also print a human friendly error from parser if the hardware cannot be found in the product list.
This commit is contained in:
1
.github/workflows/boolean_benchmark.yml
vendored
1
.github/workflows/boolean_benchmark.yml
vendored
@@ -90,6 +90,7 @@ jobs:
|
||||
- name: Parse AVX512 results
|
||||
run: |
|
||||
python3 ./ci/benchmark_parser.py target/criterion ${{ env.RESULTS_FILENAME }} \
|
||||
--hardware ${{ inputs.instance_type }} \
|
||||
--name-suffix avx512 \
|
||||
--throughput \
|
||||
--append-results
|
||||
|
||||
1
.github/workflows/shortint_benchmark.yml
vendored
1
.github/workflows/shortint_benchmark.yml
vendored
@@ -91,6 +91,7 @@ jobs:
|
||||
- name: Parse AVX512 results
|
||||
run: |
|
||||
python3 ./ci/benchmark_parser.py target/criterion ${{ env.RESULTS_FILENAME }} \
|
||||
--hardware ${{ inputs.instance_type }} \
|
||||
--walk-subdirs \
|
||||
--name-suffix avx512 \
|
||||
--throughput \
|
||||
|
||||
@@ -227,8 +227,12 @@ if __name__ == "__main__":
|
||||
print("Throughput computation enabled")
|
||||
ec2_costs = json.loads(
|
||||
pathlib.Path("ci/ec2_products_cost.json").read_text(encoding="utf-8"))
|
||||
hardware_cost = abs(ec2_costs[args.hardware])
|
||||
print(f"Hardware hourly cost: {hardware_cost} $/h")
|
||||
try:
|
||||
hardware_cost = abs(ec2_costs[args.hardware])
|
||||
print(f"Hardware hourly cost: {hardware_cost} $/h")
|
||||
except KeyError:
|
||||
print(f"Cannot find hardware hourly cost for '{args.hardware}'")
|
||||
sys.exit(1)
|
||||
|
||||
results = recursive_parse(raw_results, args.walk_subdirs, args.name_suffix, args.throughput,
|
||||
hardware_cost)
|
||||
|
||||
Reference in New Issue
Block a user