Skip to content

Commit fc7a832

Browse files
committed
Simplify getTerminatingOutputs
1 parent 71f6b1c commit fc7a832

File tree

1 file changed

+3
-14
lines changed

1 file changed

+3
-14
lines changed

torch/csrc/jit/codegen/cuda/fusion.cpp

Lines changed: 3 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -562,24 +562,13 @@ std::vector<Val*> Fusion::getTerminatingOutputs() {
562562
used_vals.emplace(inp);
563563
}
564564

565-
std::unordered_set<Val*> terminating_outputs;
565+
std::vector<Val*> terminating_outputs;
566566
for (auto out : outputs()) {
567567
if (used_vals.find(out) != used_vals.end())
568568
continue;
569-
terminating_outputs.emplace(out);
569+
terminating_outputs.push_back(out);
570570
}
571-
572-
std::vector<Val*> sorted_outputs{terminating_outputs.begin(),
573-
terminating_outputs.end()};
574-
575-
// Sort the outputs in order to give a deterministic traversal
576-
// order.
577-
std::sort(
578-
sorted_outputs.begin(),
579-
sorted_outputs.end(),
580-
[](const Val* v0, const Val* v1) { return v0->name() < v1->name(); });
581-
582-
return sorted_outputs;
571+
return terminating_outputs;
583572
}
584573

585574
} // namespace fuser

0 commit comments

Comments
 (0)