We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0d5db82 commit 791f962Copy full SHA for 791f962
src/nncf/common/insertion_point_graph.py
@@ -236,9 +236,10 @@ def _get_default_pre_hook_ip_list(nncf_graph: NNCFGraph) -> list[PreHookInsertio
236
237
for pred_node in pred_nodes:
238
input_edge = nncf_graph.get_edge(pred_node, nncf_node)
239
- allowed_pre_hook_insertion_points.append(
240
- PreHookInsertionPoint(nncf_node.node_name, input_edge.input_port_id)
241
- )
+ input_port_ids = [input_edge.input_port_id] + input_edge.parallel_input_port_ids
+ node_name = nncf_node.node_name
+ for input_port_id in input_port_ids:
242
+ allowed_pre_hook_insertion_points.append(PreHookInsertionPoint(node_name, input_port_id))
243
return allowed_pre_hook_insertion_points
244
245
@staticmethod
0 commit comments