Fix (float/clamp): Bugfix when unsigned

Xilinx · Dec 16, 2024 · 248d83a · 248d83a
1 parent 482531c
commit 248d83a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/core/function_wrapper/clamp.py b/src/brevitas/core/function_wrapper/clamp.py
@@ -152,7 +152,7 @@ def forward(
         # Compute masks
         inf_mask = x.isinf()
         p_max_val_mask = x > max_value
-        n_max_val_mask = -x > max_value
+        n_max_val_mask = x < min_value
 
         # first clamp everything to +- max_value, basically the saturating case
         x = self.saturating_clamp(x, max_value, min_value)