From 8cf75b7fe7911f542f85a66399e3515153d4f082 Mon Sep 17 00:00:00 2001 From: yuwenzho Date: Wed, 23 Nov 2022 09:27:25 +0800 Subject: [PATCH] fix topological sort assert error Signed-off-by: yuwenzho --- neural_compressor/model/onnx_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/neural_compressor/model/onnx_model.py b/neural_compressor/model/onnx_model.py index bc7d7aac987..90fcda508c6 100644 --- a/neural_compressor/model/onnx_model.py +++ b/neural_compressor/model/onnx_model.py @@ -403,7 +403,8 @@ def topological_sort(self, enable_subgraph=False): q = copy.deepcopy(wait) wait.clear() nodes = [i[1] for i in all_nodes.items()] - assert len(nodes) == len(self.model.graph.node) + assert len(list(set([n.name for n in nodes]))) == \ + len(list(set([n.name for n in self.model.graph.node]))) self.model.graph.ClearField('node') self.model.graph.node.extend(nodes)