fix: Update logging scheme for converters

- Add debug statements for converter registrations in both FX and Dynamo

Author: gs-olive

py/torch_tensorrt/dynamo/backend/lowering/_partition.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Callable, Dict, List, Optional, Sequence, Set
+from typing import Dict, List, Optional, Sequence, Set
 
 import torch
 
@@ -55,10 +55,6 @@ def __init__(
         )
 
         self.min_block_size = min_block_size
-        logger.debug(
-            "Initialized Capability-Based Partitioner with available Converters:\n"
-            + f"{CONVERTERS.display_all_available_converters()}"
-        )
 
     def propose_partitions(self) -> List[Partition]:
         # Propose partitions using the default, then refine the results

py/torch_tensorrt/dynamo/converter_registry.py

@@ -1,3 +1,4 @@
+import logging
 from dataclasses import dataclass, field
 from typing import Any, Callable, Dict, Optional, Sequence, Union
 from enum import Enum, auto
@@ -6,6 +7,9 @@
 from torch_tensorrt.fx.converter_registry import CONVERTERS
 
 
+logger = logging.getLogger(__name__)
+
+
 class ConverterPriority(Enum):
     """Enum to set a converter's priority in the registry"""
 
@@ -85,6 +89,10 @@ def register_converter(converter):
         else:
             DYNAMO_ATEN_CONVERTERS[key] = [converter_support]
 
+        logger.debug(
+            f"Converter for {key} added to Dynamo ATen Converter Registry with priority: {priority}"
+        )
+
         return converter
 
     def disable_converter(converter):

py/torch_tensorrt/fx/converter_registry.py

@@ -1,3 +1,4 @@
+import logging
 from typing import Any, Callable, Dict
 
 from torch.fx.node import Target
@@ -7,6 +8,9 @@
 NO_EXPLICIT_BATCH_DIM_SUPPORT = {}
 
 
+logger = logging.getLogger(__name__)
+
+
 def tensorrt_converter(
     key: Target,
     no_implicit_batch_dim: bool = False,
@@ -19,6 +23,13 @@ def register_converter(converter):
             NO_IMPLICIT_BATCH_DIM_SUPPORT[key] = converter
         if no_explicit_batch_dim:
             NO_EXPLICIT_BATCH_DIM_SUPPORT[key] = converter
+
+        logger.debug(
+            f"Converter for {key} added to FX Converter Registry "
+            + f"{'without' if no_explicit_batch_dim else 'with'} Explicit Batch Dim Support + "
+            + f"{'without' if no_implicit_batch_dim else 'with'} Implicit Batch Dim Support"
+        )
+
         return converter
 
     def disable_converter(converter):