fix: erase the repetitive nodes in dependency analysis

bowang007 · bowang007 · commit 80b1038cd057 · 2021-04-27T16:25:26.000-05:00
Signed-off-by: Bo Wang &lt;wangbo1995ee@163.com&gt;
diff --git a/core/partitioning/partitioning.cpp b/core/partitioning/partitioning.cpp
@@ -97,12 +97,14 @@ std::vector<SegmentedBlock> injectNodesForNonTensorInputs(SegmentedBlock& seg_bl
       if (containNonTensorInputs(n, nontensor_inputs_set) || prev_non_tensor_outputs) {
         if (!tensorrt_nodes.empty()) {
           new_seg_blocks.emplace_back(SegmentedBlock::kTensorRT, tensorrt_nodes);
+          tensorrt_nodes.clear();
         }
         pytorch_nodes.push_back(n);
         prev_non_tensor_outputs = containNonTensorOutputs(n);
       } else {
         if (!pytorch_nodes.empty()) {
           new_seg_blocks.emplace_back(SegmentedBlock::kTorch, pytorch_nodes);
+          pytorch_nodes.clear();
         }
         tensorrt_nodes.push_back(n);
       }

Original file line number	Diff line number	Diff line change
`@@ -97,12 +97,14 @@ std::vector<SegmentedBlock> injectNodesForNonTensorInputs(SegmentedBlock& seg_bl`
`97`	`97`	`if (containNonTensorInputs(n, nontensor_inputs_set) \|\| prev_non_tensor_outputs) {`
`98`	`98`	`if (!tensorrt_nodes.empty()) {`
`99`	`99`	`new_seg_blocks.emplace_back(SegmentedBlock::kTensorRT, tensorrt_nodes);`
	`100`	`+ tensorrt_nodes.clear();`
`100`	`101`	`}`
`101`	`102`	`pytorch_nodes.push_back(n);`
`102`	`103`	`prev_non_tensor_outputs = containNonTensorOutputs(n);`
`103`	`104`	`} else {`
`104`	`105`	`if (!pytorch_nodes.empty()) {`
`105`	`106`	`new_seg_blocks.emplace_back(SegmentedBlock::kTorch, pytorch_nodes);`
	`107`	`+ pytorch_nodes.clear();`
`106`	`108`	`}`
`107`	`109`	`tensorrt_nodes.push_back(n);`
`108`	`110`	`}`