From 32a70a2d8c42b730ceb11a5af4723473dee55dd9 Mon Sep 17 00:00:00 2001 From: fanyunqian Date: Mon, 11 Apr 2022 17:06:35 +0800 Subject: [PATCH] [Fix] some nodes' target is not a module. --- mqbench/prepare_by_platform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mqbench/prepare_by_platform.py b/mqbench/prepare_by_platform.py index 4e69028..24eaba2 100644 --- a/mqbench/prepare_by_platform.py +++ b/mqbench/prepare_by_platform.py @@ -296,7 +296,7 @@ def duplicate_reused_nodes(graph: torch.fx.Graph, modules: Dict[str, Any] = {}): target_dict = dict() dup_modules = dict() for node in graph.nodes: - if isinstance(node.target, str): + if node.op == "call_module": if node.target not in target_dict: target_dict[node.target] = [node] else: