diff --git a/exir/backend/backend_api.py b/exir/backend/backend_api.py index d5bd574ec5a..ab2e66f7885 100644 --- a/exir/backend/backend_api.py +++ b/exir/backend/backend_api.py @@ -401,6 +401,11 @@ def to_backend( tagged_exported_program, ) + # Partitioner added delegation tags to the graph module nodes, + # we make sure to remove them after we finished partition_and_lower + for node in tagged_graph_module.graph.nodes: + node.meta.pop("delegation_tag", None) + return ExportedProgram( root=tagged_graph_module, graph=tagged_graph_module.graph,