From b99e4a2825516add1d5254be96184cb1a3cc62d2 Mon Sep 17 00:00:00 2001 From: Bo Wun Cheng Date: Wed, 22 Nov 2023 14:41:17 -0800 Subject: [PATCH] add 'exp' to the list of glb name to be annotated --- sam/onyx/parse_dot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sam/onyx/parse_dot.py b/sam/onyx/parse_dot.py index 8533c9a8..14adb80d 100644 --- a/sam/onyx/parse_dot.py +++ b/sam/onyx/parse_dot.py @@ -1016,7 +1016,7 @@ def duplicate_graph(self, tensor, factor, output='x'): def annotate_IO_nodes(self): original_nodes = self.graph.get_nodes() output_nodes = ['x', 'X'] - input_nodes = ['c', 'C', 'b', 'B', 'd', 'D', 'e', 'E', 'f', 'F'] + input_nodes = ['c', 'C', 'b', 'B', 'd', 'D', 'e', 'E', 'f', 'F', 'exp'] exclude_nodes = ['b', 'B'] for node in original_nodes: node_attrs = node.get_attributes()