File tree Expand file tree Collapse file tree 1 file changed +3
-14
lines changed
torch/csrc/jit/codegen/cuda Expand file tree Collapse file tree 1 file changed +3
-14
lines changed Original file line number Diff line number Diff line change @@ -562,24 +562,13 @@ std::vector<Val*> Fusion::getTerminatingOutputs() {
562
562
used_vals.emplace (inp);
563
563
}
564
564
565
- std::unordered_set <Val*> terminating_outputs;
565
+ std::vector <Val*> terminating_outputs;
566
566
for (auto out : outputs ()) {
567
567
if (used_vals.find (out) != used_vals.end ())
568
568
continue ;
569
- terminating_outputs.emplace (out);
569
+ terminating_outputs.push_back (out);
570
570
}
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;
583
572
}
584
573
585
574
} // namespace fuser
You can’t perform that action at this time.
0 commit comments