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 6aee084 commit bf6645fCopy full SHA for bf6645f
core/partitioning/partitioning.cpp
@@ -178,7 +178,7 @@ void resolveTRTNonTensorInputs(PartitionedGraph& segmented_blocks) {
178
179
void registerSegmentsOutputs(PartitionedGraph& segmented_blocks, torch::jit::Block* block) {
180
// find the corresponding raw values in original global graph for this segmented block's inputs/outputs
181
- auto cmp = [](torch::jit::Value* a, torch::jit::Value* b) { return a->unique() < b->unique();};
+ auto cmp = [](torch::jit::Value* a, torch::jit::Value* b) { return a->unique() < b->unique(); };
182
std::set<torch::jit::Value*, decltype(cmp)> input_values(cmp);
183
for (auto& seg_block : segmented_blocks) {
184
for (auto& input : seg_block.raw_inputs()) {
0 commit comments