docstring

jenchen13 · jenchen13 · commit d1fac44cd782 · 2025-10-01T20:58:09.000Z
Signed-off-by: Jennifer Chen &lt;jennifchen@nvidia.com&gt;
diff --git a/modelopt/torch/quantization/model_calib.py b/modelopt/torch/quantization/model_calib.py
@@ -117,6 +117,7 @@ def sync_quantizer_amax_across_tp(
     ):
         if isinstance(quantizer, SequentialQuantizer):
             for _q in quantizer:
+                "Syncing amax across TP for sequential quantizer"
                 sync_quantizer_amax_across_tp(
                     _q, linear_name, quantizer_type, axes_for_sync, parallel_state
                 )

Original file line number	Diff line number	Diff line change
`@@ -117,6 +117,7 @@ def sync_quantizer_amax_across_tp(`
`117`	`117`	`):`
`118`	`118`	`if isinstance(quantizer, SequentialQuantizer):`
`119`	`119`	`for _q in quantizer:`
	`120`	`+ "Syncing amax across TP for sequential quantizer"`
`120`	`121`	`sync_quantizer_amax_across_tp(`
`121`	`122`	`_q, linear_name, quantizer_type, axes_for_sync, parallel_state`
`122`	`123`	`)`