From 31f69ede4f4c2223e756298a54184a5d7597a99e Mon Sep 17 00:00:00 2001 From: Ganesan Ramalingam Date: Thu, 5 Sep 2024 14:20:47 -0700 Subject: [PATCH] Use ir_convenience --- onnxscript/optimizer/_inliner.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/onnxscript/optimizer/_inliner.py b/onnxscript/optimizer/_inliner.py index cea91c7e2..6a6cd9bb3 100644 --- a/onnxscript/optimizer/_inliner.py +++ b/onnxscript/optimizer/_inliner.py @@ -9,7 +9,7 @@ from typing import Iterable, Sequence, Tuple import onnxscript.ir as ir -import onnxscript.ir.convenience as convenience +import onnxscript.ir.convenience as ir_convenience # A replacement for a node specifies a list of nodes that replaces the original node, # and a list of values that replaces the original node's outputs. @@ -115,7 +115,7 @@ def clone_node(self, node: ir.Node) -> ir.Node: new_attributes, overload=node.overload, num_outputs=len(node.outputs), - graph=None, # TODO: + graph=None, name=new_name, doc_string=node.doc_string, metadata_props=new_metadata, @@ -252,7 +252,7 @@ def inline_calls_in(self, graph: ir.Graph) -> None: call_site_prefix = "" call_site = node.name or (self._function_id_abbreviations[id] + call_site_prefix) nodes, values = self._instantiate_call(node, call_site) - convenience.replace_nodes_and_values( + ir_convenience.replace_nodes_and_values( graph, insertion_point=node, old_nodes=[node], new_nodes=nodes, old_values=node.outputs, new_values=values ) else: