Fix tests

Xilinx · Jan 21, 2024 · 2ce9fdf · 2ce9fdf
1 parent ef83ffb
commit 2ce9fdf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/brevitas_examples/common/generative/quantizers.py b/src/brevitas_examples/common/generative/quantizers.py
@@ -131,7 +131,6 @@ class Int8ActDynamicPerTensorFloat(ActDynamicProxyMixin, Int8ActPerTensorFloat):
     scaling_stats_input_view_shape_impl = OverTensorView
     scaling_stats_op = 'min_max'
     dynamic_scaling_broadcastable_shape = this.scaling_shape
-    stats_reduce_dim = 0
 
 
 class Int8ActDynamicPerRowFloat(ActDynamicProxyMixin, Int8ActPerRowFloat):
@@ -165,4 +164,3 @@ class ShiftedUint8ActDynamicPerTensorFloat(ActDynamicProxyMixin, ShiftedUint8Act
     scaling_stats_op = 'min_max'
     zero_point_stats_impl = NegativeMinOrZero
     dynamic_scaling_broadcastable_shape = this.scaling_shape
-    stats_reduce_dim = 0
diff --git a/tests/brevitas_ort/common.py b/tests/brevitas_ort/common.py
@@ -58,7 +58,9 @@ class A2QWeightQuantizerForTests(Int8AccumulatorAwareWeightQuant):
     'a2q': (A2QWeightQuantizerForTests, Int8ActPerTensorFloat),
     'symmetric_per_tensor_fixed_point': (Int8WeightPerTensorFixedPoint, Int8ActPerTensorFixedPoint),
     'symmetric_per_channel_fixed_point':
-        (Int8WeightPerChannelFixedPoint, Int8ActPerTensorFixedPoint)}
+        (Int8WeightPerChannelFixedPoint, Int8ActPerTensorFixedPoint),
+    'weight_symmetric_activation_dynamic_asymmetric_per_tensor_float':
+        (Int8WeightPerTensorFloat, ShiftedUint8ActDynamicPerTensorFloat)}
 LSTM_QUANTIZERS = {
     'asymmetric_per_tensor_float':
         (ShiftedUint8WeightPerTensorFloat, ShiftedUint8ActPerTensorFloat),