Skip to content

GitLab

Unverified Commit 681ccc59 authored by Ningxin Zheng's avatar Ningxin Zheng Committed by GitHub
Browse files

remove pdb (#4151)

parent 396ae65c
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 6 deletions
nni/retiarii/converter/graph_gen.py
View file @ 681ccc59
...@@ -818,12 +818,10 @@ class GraphConverterWithShape(GraphConverter): ...@@ -818,12 +818,10 @@ class GraphConverterWithShape(GraphConverter):
graph.del_edge(out_edge) graph.del_edge(out_edge)
for output_node_edge in node_graph.output_node.incoming_edges: for output_node_edge in node_graph.output_node.incoming_edges:
if output_node_edge.head_slot == out_edge.tail_slot: if output_node_edge.head_slot == out_edge.tail_slot:
try:
graph.add_edge( graph.add_edge(
head=(id_to_new_node[output_node_edge.head.id], output_node_edge.head_slot), head=(id_to_new_node[output_node_edge.head.id], output_node_edge.head_slot),
tail=(out_edge.tail, out_edge.tail_slot)) tail=(out_edge.tail, out_edge.tail_slot))
except:
import pdb; pdb.set_trace()
for edge in node_graph.edges: for edge in node_graph.edges:
if edge.head == node_graph.input_node or edge.tail == node_graph.output_node: if edge.head == node_graph.input_node or edge.tail == node_graph.output_node:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment