Skip to content

Commit 6f675f3

Browse files
authored
[backend_api] Delete partitioner tags after lowering
Differential Revision: D71518362 Pull Request resolved: #9435
1 parent 17c08c2 commit 6f675f3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

exir/backend/backend_api.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -401,6 +401,11 @@ def to_backend(
401401
tagged_exported_program,
402402
)
403403

404+
# Partitioner added delegation tags to the graph module nodes,
405+
# we make sure to remove them after we finished partition_and_lower
406+
for node in tagged_graph_module.graph.nodes:
407+
node.meta.pop("delegation_tag", None)
408+
404409
return ExportedProgram(
405410
root=tagged_graph_module,
406411
graph=tagged_graph_module.graph,

0 commit comments

Comments
 (0)