Remove the used metadata from the graph.
(graph: torch.fx.Graph)
| 641 | |
| 642 | |
| 643 | def remove_used_metadata(graph: torch.fx.Graph) -> None: |
| 644 | """ |
| 645 | Remove the used metadata from the graph. |
| 646 | """ |
| 647 | for node in graph.nodes: |
| 648 | node.meta.pop("delegation_tag", None) |
| 649 | node.meta.pop("backend_id", None) |
| 650 | node.meta.pop("submodule_program", None) |
| 651 | node.meta.pop("toplevel_input_specs_to_delete", None) |
| 652 | node.meta.pop("toplevel_output_specs_to_delete", None) |
| 653 | node.meta.pop("is_submodule", None) |
| 654 | node.meta.pop("submodule_output_node", None) |
| 655 | |
| 656 | |
| 657 | @dataclass |