Removed redundant comments

Signed-off-by: greg-kwasniewski1 <[email protected]>

Author: greg-kwasniewski1

tensorrt_llm/_torch/auto_deploy/transformations/library/sharding.py

@@ -718,7 +718,6 @@ def detect_column_row_shard(
 
     if sharding_config.use_sharding_from_factory and sharding_config.predefined_config is not None:
         ad_logger.info("Using TP sharding from config")
-        # sharding_config.simple_shard_attention_layers()
         detect_tp_sharding_from_factory_config(gm, sharding_config)
         return
 

tensorrt_llm/_torch/auto_deploy/transformations/transform.py

@@ -131,13 +131,6 @@ def __call__(self, cm: CachedSequenceInterface) -> nn.Module:
         # run BMM sharding across ranks
         detect_dp_bmm_shard(egm, sharding_config)
 
-        # print detected transformations
-        ad_logger.info("\n\nTP sharding:")
-        for tp_transform in sharding_config.tp_transforms:
-            ad_logger.info(
-                f"{tp_transform.target_node} {tp_transform.split_dim} {tp_transform.dist_op}"
-            )
-
         sharding_transform_executor(egm, sharding_config)
 
         # let's run a shape propagation pass to update the graph with correct meta values for