Skip to content

Commit c3d90eb

Browse files
authored
fix autoprecision fallback for generic weights (#1453)
1 parent 90eea33 commit c3d90eb

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

hls4ml/model/optimizer/passes/infer_precision.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,9 @@ def _infer_default_type(self, node, type_name):
118118
default_precision = model_config.backend.convert_precision_string(model_config.model_precision['default'])
119119
# No need to change the name of the NamedType since we use the default precision
120120
node.types[type_name].precision = default_precision
121+
weight_name = type_name.removesuffix('_t')
122+
if weight_name in node.weights:
123+
node.weights[weight_name].update_precision(default_precision)
121124

122125
def _infer_output_matching_precision(self, node, types_to_infer):
123126
assert 'result_t' in types_to_infer and len(types_to_infer) == 1

0 commit comments

Comments
 (0)