Skip to content

Commit 58df59a

Browse files
committed
Convert segmentBlocksWithSpecifiedInputs input vector to a const ref
Signed-off-by: Michael Feliz <[email protected]>
1 parent 3e090ee commit 58df59a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

core/partitioning/partitioning.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ bool containNonTensorOutputs(torch::jit::Node* n) {
5757
return false;
5858
}
5959

60-
std::vector<torch::jit::Node*> getDependencyNodes(std::vector<torch::jit::Value*>& vals) {
60+
std::vector<torch::jit::Node*> getDependencyNodes(const std::vector<torch::jit::Value*>& vals) {
6161
// use bfs to get the DAG dependency nodes for input value
6262
std::queue<torch::jit::Value*, std::deque<torch::jit::Value*>> q(
6363
std::deque<torch::jit::Value*>(vals.begin(), vals.end()));
@@ -137,7 +137,7 @@ std::pair<std::unordered_map<torch::jit::Value*, SegmentedBlock>, SegmentedBlock
137137
return std::pair<std::unordered_map<torch::jit::Value*, SegmentedBlock>, SegmentedBlock>(append_blocks, trt_block);
138138
}
139139

140-
PartitionedGraph segmentBlocksWithSpecifiedInputs(SegmentedBlock& seg_block, std::vector<torch::jit::Value*> inputs_to_resolve){
140+
PartitionedGraph segmentBlocksWithSpecifiedInputs(SegmentedBlock& seg_block, const std::vector<torch::jit::Value*> &inputs_to_resolve){
141141
std::vector<torch::jit::Node*> dependency_nodes = getDependencyNodes(inputs_to_resolve);
142142
PartitionedGraph new_seg_blocks;
143143
// if current block is kTorch or current block is TensorRT and all dependent nodes are also supported, merge the

0 commit comments

Comments
 (0)