Update _fx_passes.py

pytorch · xadupre · Oct 7, 2024 · Oct 7, 2024 · Nov 4, 2024 · Nov 4, 2024
commit 600e23e1c5570e8f18c7e61c500dc198c1278562
@@ -23,7 +23,7 @@ def decompose_with_registry(
 def insert_type_promotion_nodes(
     graph_module: torch.fx.GraphModule,
 ) -> None:
-    """Inplace pass to insert explicit type promotion nodes."""
+    """Inplace pass to insert explicit type promotion nodes, recursively through nested modules."""
     for module in graph_module.modules():
         assert isinstance(module, torch.fx.GraphModule)
         diagnostic_context = diagnostics.DiagnosticContext(