8000
We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 11ad928 commit eb54774Copy full SHA for eb54774
torch/_inductor/fx_passes/micro_pipeline_tp.py
@@ -855,9 +855,7 @@ def _get_node_to_ancestors(
855
"""
856
Compute the ancestors for all nodes in a graph.
857
858
- node_to_ancestors = defaultdict(
859
- OrderedSet[torch.fx.Node]
860
- ) # type: ignore[var-annotated]
+ node_to_ancestors = defaultdict(OrderedSet[torch.fx.Node]) # type: ignore[var-annotated]
861
for node in graph.nodes:
862
node_to_ancestors[node] = OrderedSet(node.all_input_nodes)
863
for dep in node.all_input_nodes:
0 commit comments