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

chore(ci): support shortint layer name parsing in data extractor

Browse Source
This commit is contained in:
David Testé
2025-10-28 10:12:52 +01:00
committed by David Testé
parent f768fd1cdd
commit 67130646ad
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ci/data_extractor/src/benchmark_specs.py
View File

@@ -493,6 +493,8 @@ class BenchDetails:
else: else:
self.operation_name = "::".join(parts[1:-1]) self.operation_name = "::".join(parts[1:-1])
self.rust_type = parts[-1].partition("_mean")[0] self.rust_type = parts[-1].partition("_mean")[0]
case Layer.Shortint:
self.operation_name = parts[1]
case _: case _:
raise NotImplementedError( raise NotImplementedError(
f"layer '{self.layer}' not supported yet for name parsing" f"layer '{self.layer}' not supported yet for name parsing"