Fix (per_group): fixing the per_group sym quantizer (#1089)

Xilinx · Nov 7, 2024 · 6c0560f · 6c0560f
1 parent 84cdfc3
commit 6c0560f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/brevitas_examples/common/generative/quantize.py b/src/brevitas_examples/common/generative/quantize.py
@@ -181,7 +181,8 @@
         'dynamic': {
             'float_scale': {
                 'stats': {
-                    'per_group': Fp8e4m3DynamicActPerGroupFloat}}},
+                    'per_group': {
+                        'sym': Fp8e4m3DynamicActPerGroupFloat}}}},
         'no_scale': {
             'sym': Fp8e4m3Act,}},
     'float_ocp': {