Fix module name extraction logic in quant_api.py

jcaip · web-flow · commit 4d1d8a3715c5 · 2025-11-05T22:41:40.000-05:00
diff --git a/torchao/quantization/quant_api.py b/torchao/quantization/quant_api.py
@@ -485,7 +485,7 @@ def quantize_(
                 or ("_default" in config.fqn_to_config and _is_linear(module))
             ):
                 module_name = (
-                    module_fqn.rsplit(".", 1) if "." in module_fqn else module_fqn
+                    module_fqn.rsplit(".", 1)[0] if "." in module_fqn else module_fqn
                 )
                 # this replaces inplace, so no need to reassign
                 _fqn_to_config_handler(module, module_name, config, device)

Original file line number	Diff line number	Diff line change
`@@ -485,7 +485,7 @@ def quantize_(`
`485`	`485`	`or ("_default" in config.fqn_to_config and _is_linear(module))`
`486`	`486`	`):`
`487`	`487`	`module_name = (`
`488`		`- module_fqn.rsplit(".", 1) if "." in module_fqn else module_fqn`
	`488`	`+ module_fqn.rsplit(".", 1)[0] if "." in module_fqn else module_fqn`
`489`	`489`	`)`
`490`	`490`	`# this replaces inplace, so no need to reassign`
`491`	`491`	`_fqn_to_config_handler(module, module_name, config, device)`