add 'exp' to the list of glb name to be annotated

weiya711 · Nov 22, 2023 · b99e4a2 · b99e4a2
1 parent 5deea43
commit b99e4a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()