fix

microsoft · Apr 18, 2021 · f31b019 · f31b019
1 parent 9c09bf0
commit f31b019
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nni/compression/pytorch/compressor.py b/nni/compression/pytorch/compressor.py
@@ -720,7 +720,7 @@ def quant_backward(tensor, grad_output, quant_type, scale, zero_point, qmin, qma
 
     @staticmethod
     def forward(ctx, tensor, quant_type, wrapper, input_tensor=None, **kwargs):
-        output = quantize_helper(tensor, quant_type, wrapper, **kwargs)
+        output = quantize_helper(tensor, quant_type, wrapper, input_tensor, **kwargs)
 
         bits = QuantGrad.get_bits_length(wrapper.config, QType_Dict[quant_type])
         qmin, qmax = torch.Tensor([0]).to(tensor.device), torch.Tensor([(1 << bits) - 1]).to(tensor.device)