Fix AMP v2 slowdown issue due to compute_param_encodings

Signed-off-by: Seokjun An <[email protected]>
quic · Jan 3, 2025 · e1836f7 · e1836f7
1 parent 421aece
commit e1836f7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/TrainingExtensions/torch/src/python/aimet_torch/v2/amp/utils.py b/TrainingExtensions/torch/src/python/aimet_torch/v2/amp/utils.py
@@ -175,6 +175,9 @@ def enabled(self, val): # pylint: disable=no-self-use
         if val:
             raise RuntimeError
 
+    def __bool__(self):
+        return False
+
 
 class _V1QuantizeDequantize(_V1QuantizerMixin, QuantizeDequantize):
     ...