Skip to content

Commit 82ef0ec

Browse files
committed
bit-exact algorithm minor change
1 parent 6873934 commit 82ef0ec

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

hls4ml/model/optimizer/passes/bit_exact.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -585,20 +585,20 @@ def request_kif(layer: Layer) -> tuple[KIF_t, ...]:
585585
return kif
586586

587587

588-
def requested_by_quantizer(layer: Layer) -> bool:
588+
def requested_by_non_saturating_quantizer(layer: Layer) -> bool:
589589
"""Check if the current requested kif is from a quantizer.
590590
591591
Args:
592592
layer (Layer): The layer to check.
593593
594594
Returns:
595-
bool: True if requested by a quantizer, False otherwise.
595+
bool: True if requested by a non-saturating quantizer, False otherwise.
596596
"""
597597
for n in get_output_layers(layer):
598-
if isinstance(n, FixedPointQuantizer):
598+
if isinstance(n, FixedPointQuantizer) and n.SAT not in ('SAT', 'SAT_SYM'):
599599
return True
600600
if isinstance(n, Reshape):
601-
return requested_by_quantizer(n)
601+
return requested_by_non_saturating_quantizer(n)
602602
return False
603603

604604

@@ -607,7 +607,7 @@ def default_register_precision(layer: Layer):
607607
_rk, _ri, _rf = requested_kif(layer) # Maximum possible k,i,f may be utilized by the next layer
608608
_oi, _of = np.minimum(_pi, _ri), np.minimum(_pf, _rf)
609609

610-
if requested_by_quantizer(layer):
610+
if requested_by_non_saturating_quantizer(layer):
611611
_ok = _rk
612612
else:
613613
_ok = np.minimum(_pk, _rk)

0 commit comments

Comments
 (0)