Fix (gpxq): removing null ops

Xilinx · Apr 26, 2024 · a24afdc · a24afdc
1 parent 2e519f5
commit a24afdc
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/src/brevitas/graph/gpfq.py b/src/brevitas/graph/gpfq.py
@@ -17,7 +17,6 @@
 from brevitas.graph.gpxq import StopFwdException
 from brevitas.graph.gpxq import SUPPORTED_CONV_OP
 import brevitas.nn as qnn
-from brevitas.quant_tensor import _unpack_quant_tensor
 
 
 class gpfq_mode(gpxq_mode):
@@ -163,7 +162,6 @@ def update_batch(self, module, input, current_layer):
         is_quant_enabled = module.weight_quant.is_quant_enabled
 
         inp = self.process_input(input)
-        inp = _unpack_quant_tensor(inp)
         batch_size = inp.shape[0]
 
         # Preprocess the input to compute the Hessian

diff --git a/src/brevitas/graph/gptq.py b/src/brevitas/graph/gptq.py
@@ -21,7 +21,6 @@
 from brevitas.graph.gpxq import StopFwdException
 from brevitas.graph.gpxq import SUPPORTED_CONV_OP
 import brevitas.nn as qnn
-from brevitas.quant_tensor import _unpack_quant_tensor
 
 
 class gptq_mode(gpxq_mode):
@@ -145,7 +144,6 @@ def update_batch(self, module, input, current_layer):
         # Update reference to current layer
         current_layer.layer_names.add(self.name)
         inp = self.process_input(input)
-        inp = _unpack_quant_tensor(inp)
         batch_size = inp.shape[0]
 
         # Preprocess the input to compute the Hessian