From 7ed146b5c7bfb141d1d32051d31cd3a5f6bdd3b0 Mon Sep 17 00:00:00 2001 From: inisis Date: Fri, 24 Jan 2025 00:23:57 +0800 Subject: [PATCH] fix: Cast elimination (#77) --- onnxslim/core/optimization/dead_node_elimination.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxslim/core/optimization/dead_node_elimination.py b/onnxslim/core/optimization/dead_node_elimination.py index 6d7276e..8bcf6aa 100644 --- a/onnxslim/core/optimization/dead_node_elimination.py +++ b/onnxslim/core/optimization/dead_node_elimination.py @@ -28,7 +28,7 @@ def dead_node_elimination(graph, is_subgraph=False): if all(value == 0 for value in pad_value): node.erase() logger.debug(f"removing {node.op} op: {node.name}") - elif node.op == "Cast_": + elif node.op == "Cast": inp_dtype = [dtype_to_onnx(input.dtype) for input in node.inputs][0] if inp_dtype == node.attrs["to"]: node.erase()