up

metascroy · metascroy · commit 5e07eb3168cd · 2025-08-17T17:02:47.000-07:00
diff --git a/torchao/quantization/quant_api.py b/torchao/quantization/quant_api.py
@@ -2060,6 +2060,7 @@ def _intx_weight_only_quantize_tensor(weight, config):
         zero_point_domain=ZeroPointDomain.INT,
         _layout=layout,
     )
+    return weight
 
 
 @register_quantize_module_handler(IntxWeightOnlyConfig)

Original file line number	Diff line number	Diff line change
`@@ -2060,6 +2060,7 @@ def _intx_weight_only_quantize_tensor(weight, config):`
`2060`	`2060`	`zero_point_domain=ZeroPointDomain.INT,`
`2061`	`2061`	`_layout=layout,`
`2062`	`2062`	`)`
	`2063`	`+ return weight`
`2063`	`2064`
`2064`	`2065`
`2065`	`2066`	`@register_quantize_module_handler(IntxWeightOnlyConfig)`