Skip to content

Commit 4a318a2

Browse files
committed
chore: apply linting
Signed-off-by: Bo Wang <[email protected]>
1 parent e491bb5 commit 4a318a2

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

core/partitioning/partitioning.cpp

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,6 @@ std::vector<SegmentedBlock> injectNodesForNonTensorInputs(SegmentedBlock& seg_bl
9090
} else {
9191
// if current block is kTensorRT but the dependency nodes contain unsupported node, then we have to segment again
9292
std::unordered_set<torch::jit::Value*> nontensor_inputs_set(nontensor_inputs.begin(), nontensor_inputs.end());
93-
new_seg_blocks.emplace_back(SegmentedBlock::kTorch, dependency_nodes);
9493
std::vector<torch::jit::Node*> tensorrt_nodes, pytorch_nodes;
9594
bool prev_non_tensor_outputs = false;
9695
for (auto n : seg_block.raw_nodes()) {
@@ -204,17 +203,16 @@ void registerSegmentsOutputs(PartitionedGraph& segmented_blocks, std::shared_ptr
204203
}
205204
}
206205
}
207-
std::for_each(
208-
segmented_blocks.begin(),
209-
segmented_blocks.end(),
210-
[](SegmentedBlock& seg_block) { torch::jit::EliminateDeadCode(seg_block.g()); });
211-
// erase segments which still have no output
212-
segmented_blocks.erase(
213-
std::remove_if(
214-
segmented_blocks.begin(),
215-
segmented_blocks.end(),
216-
[](SegmentedBlock& seg_block) { return seg_block.raw_outputs().empty(); }),
217-
segmented_blocks.end());
206+
std::for_each(segmented_blocks.begin(), segmented_blocks.end(), [](SegmentedBlock& seg_block) {
207+
torch::jit::EliminateDeadCode(seg_block.g());
208+
});
209+
// erase segments which still have no output
210+
segmented_blocks.erase(
211+
std::remove_if(
212+
segmented_blocks.begin(),
213+
segmented_blocks.end(),
214+
[](SegmentedBlock& seg_block) { return seg_block.raw_outputs().empty(); }),
215+
segmented_blocks.end());
218216

219217
return;
220218
}

0 commit comments

Comments
 (0)