rename

pytorch · Apr 12, 2023 · 7d739a6 · 7d739a6
1 parent ca24bad
commit 7d739a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch/fx/passes/utils/source_matcher_utils.py b/torch/fx/passes/utils/source_matcher_utils.py
@@ -7,7 +7,7 @@
 import os
 
 
-__all__ = ['get_module_partitions', 'check_subgraphs_connected']
+__all__ = ['get_source_partitions', 'check_subgraphs_connected']
 
 # Set`PYTORCH_MATCHER_LOGLEVEL=INFO` to see debug logs
 def _init_logger():
@@ -60,7 +60,7 @@ def get_source_partitions(
             leaf module type (ex. torch.nn.Linear).
 
     Returns:
-        Dictionary mapping sources that were given to a list of ModulePartitions
+        Dictionary mapping sources that were given to a list of SourcePartitions
         that correspond to the list of nodes that were decomposed from the given
         source.
     """