onnx-graphsurgeon: fix constant folding bug when input is empty

Signed-off-by: inisis <[email protected]>
NVIDIA · Oct 14, 2023 · 3aaa97b · 3aaa97b
1 parent 78245b0
commit 3aaa97b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/onnx-graphsurgeon/onnx_graphsurgeon/ir/graph.py b/tools/onnx-graphsurgeon/onnx_graphsurgeon/ir/graph.py
@@ -668,7 +668,7 @@ def is_foldable(node):
                     return False
 
                 def all_tensors_const(tensors):
-                    return all([t.name in graph_constants for t in tensors])
+                    return all([t.name in graph_constants for t in tensors if not t.is_empty()])
 
                 if not all_tensors_const(node.inputs):
                     return False