Added tests for symm/asym quantization, per_chan quantization, and st…

…arted work on minifloat quantization
Xilinx · Apr 14, 2024 · b0d1a7d · b0d1a7d
1 parent 56daf8d
commit b0d1a7d
Show file tree

Hide file tree

Showing 2 changed files with 465 additions and 110 deletions.
diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
@@ -557,6 +557,8 @@ def check_positive_int(*args):
     We check that every inputted value is positive, and an integer.
     """
     for arg in args:
+        if not arg:
+            continue
         assert arg > 0.0
         assert not math.isclose(arg, 0.0)
         assert math.isclose(arg % 1, 0.0)