File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -69,7 +69,7 @@ def __init__(
6969 )
7070 self .dependency_viewer = _DependencyViewer (graph_module )
7171
72- def __is_node_supported (self , node : Node ) -> bool :
72+ def _is_node_supported (self , node : Node ) -> bool :
7373 return self .operator_support .is_node_supported (
7474 dict (self .graph_module .named_modules ()), node
7575 )
@@ -206,7 +206,7 @@ def _update_partition_map(node: Node, id: int):
206206 #
207207 # I don't see a need to add a knob to disable horizontal fusion yet, we can short-cut
208208 # the fusion by adding an `else` block here to skip horizontal fusion.
209- if self .__is_node_supported (node ) and node not in assignment :
209+ if self ._is_node_supported (node ) and node not in assignment :
210210 partition_id = next (new_partition_id )
211211 nodes_order [node ] = partition_id
212212 partitions_order [partition_id ] = partition_id
You can’t perform that action at this time.
0 commit comments