Skip to content

Commit b16c0d6

Browse files
StellarrZamathewc
authored andcommitted
[Sigmoid] Remove magic method in CapabilityBasedPartitioner (pytorch#149400)
Summary: As title. Test Plan: CI Differential Revision: D70575197 Pull Request resolved: pytorch#149400 Approved by: https://github.com/jfix71
1 parent f3bb868 commit b16c0d6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torch/fx/passes/infra/partitioner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)