fix init

Xilinx · Aug 25, 2024 · f2924a5 · f2924a5
1 parent ed87026
commit f2924a5
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/src/brevitas/nn/mixin/base.py b/src/brevitas/nn/mixin/base.py
@@ -17,6 +17,7 @@
     is_dynamo_compiling = lambda: False
 else:
     is_dynamo_compiling = torch._dynamo.is_compiling
+
 from torch import nn
 from torch import Tensor
 import torch.jit

diff --git a/src/brevitas/proxy/float_runtime_quant.py b/src/brevitas/proxy/float_runtime_quant.py
@@ -64,7 +64,7 @@ def is_fnuz(self):
 class ActFloatQuantProxyFromInjector(ActFloatQuantProxyFromInjectorBase):
 
     def __init__(self, quant_layer, quant_injector):
-        super().__init__(self, quant_layer, quant_injector)
+        super().__init__(quant_layer, quant_injector)
         self.cache_class = _CachedIOFloat
 
     def create_quant_tensor(self, qt_args, x=None):

diff --git a/src/brevitas/proxy/groupwise_float_runtime_quant.py b/src/brevitas/proxy/groupwise_float_runtime_quant.py
@@ -6,7 +6,7 @@
 class GroupwiseActFloatQuantProxyFromInjector(ActFloatQuantProxyFromInjectorBase):
 
     def __init__(self, quant_layer, quant_injector):
-        super().__init__(self, quant_layer, quant_injector)
+        super().__init__(quant_layer, quant_injector)
         self.cache_class = _CachedIOGroupwiseFloat
 
     @property

diff --git a/src/brevitas/proxy/groupwise_int_runtime_quant.py b/src/brevitas/proxy/groupwise_int_runtime_quant.py
@@ -12,7 +12,7 @@
 class GroupwiseActQuantProxyFromInjector(ActQuantProxyFromInjector):
 
     def __init__(self, quant_layer, quant_injector):
-        super().__init__(self, quant_layer, quant_injector)
+        super().__init__(quant_layer, quant_injector)
         self.cache_class = _CachedIOGroupwiseInt
 
     @property