fix tensor copy bug and skip some constant folding (#61)

inisis · Dec 18, 2024 · 9a71a41 · 9a71a41
1 parent 7ff3969
commit 9a71a41
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/onnxslim/third_party/onnx_graphsurgeon/ir/graph.py b/onnxslim/third_party/onnx_graphsurgeon/ir/graph.py
@@ -915,6 +915,7 @@ def is_foldable(node):
                     "QuantizeLinear",
                     "DequantizeLinear",
                     "DynamicQuantizeLinear",
+                    "SequenceEmpty",
                 }
                 if node.op in NO_FOLD_OPS:
                     return False

diff --git a/onnxslim/third_party/onnx_graphsurgeon/ir/tensor.py b/onnxslim/third_party/onnx_graphsurgeon/ir/tensor.py
@@ -236,7 +236,7 @@ def copy(self):
 
         Note: Generally, you should only ever make a copy of a Graph.
         """
-        return Variable(self.name, self.dtype, self.shape)
+        return Variable(self.name, self.dtype, self.shape, self.type)
 
     def __eq__(self, other):
         """Perform a check to see if two variables are equal."""