github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(frontend-python): add show assigned bit widths option to graph formatting

Browse Source
This commit is contained in:
Umut
2023-04-19 12:02:43 +02:00
committed by Quentin Bourgerie
parent efd40f9904
commit 7ce6f962e0
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
frontends/concrete-python/concrete/fhe/representation/graph.py
View File

@@ -195,6 +195,7 @@ class Graph:
show_bounds: bool = True,
show_tags: bool = True,
show_locations: bool = False,
show_assigned_bit_widths: bool = False,
) -> str:
"""
Get the textual representation of the `Graph`.
@@ -218,6 +219,9 @@ class Graph:
show_locations (bool, default = False):
whether to show line information of nodes
show_assigned_bit_widths (bool, default = False)
whether to show assigned bit width of nodes instead of their original bit width
Returns:
str:
textual representation of the `Graph`
@@ -297,7 +301,11 @@ class Graph:
bounds += "]"
output_value = node.output
if isinstance(output_value.dtype, Integer) and "original_bit_width" in node.properties:
if (
not show_assigned_bit_widths
and isinstance(output_value.dtype, Integer)
and "original_bit_width" in node.properties
):
output_value = deepcopy(output_value)
output_value.dtype.bit_width = node.properties["original_bit_width"]