Fix the test case

Signed-off-by: Jingyu Xin <[email protected]>

Author: jingyu-ml

modelopt/torch/quantization/tensor_quant.py

@@ -468,7 +468,7 @@ def forward(
     @staticmethod
     def backward(ctx, grad_outputs):
         """Implements straight through estimation with clipping."""
-        return _fake_quant_backward_function(ctx, grad_outputs, num_args=7)
+        return _fake_quant_backward_function(ctx, grad_outputs, num_args=8)
 
 
 def _dynamic_block_quantize_forward(