Skip to content

Commit 6829897

Browse files
digantdesaipytorchmergebot
authored andcommitted
Remove assert from partitioner.py (pytorch#143376)
Remove erroneous assert assuming a dependent (user) node to be in the partition. This partially reverts pytorch#136616 by removing the assert. Tested locally with a failing ExecuTorch Arm test using ``` $ python -m examples.arm.aot_arm_compiler --model_name mv2 --target ethos-u55-128 --delegate --quantize ``` Pull Request resolved: pytorch#143376 Approved by: https://github.com/tarun292
1 parent 6715a88 commit 6829897

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

torch/fx/passes/infra/partitioner.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -182,11 +182,6 @@ def _update_partition_map(node: Node, id: int):
182182
if target_id is not None:
183183
partition_map[id].add(target_id)
184184
partition_map[id].update(partition_map[target_id])
185-
else:
186-
assert not self.__is_node_supported(
187-
user_node
188-
), "Encountered user node which has not been traversed yet. \
189-
This should only happen if this is an unsupported node."
190185

191186
# Iterate through all the upstream nodes of this node and update the partition map
192187
# to indicate that there is a path from the partition id of the upstream node to the

0 commit comments

Comments
 (0)