Commit 8c1231e8 authored by Paul's avatar Paul
Browse files

Formatting

parent 3b800ff3
...@@ -57,7 +57,7 @@ struct stream_info ...@@ -57,7 +57,7 @@ struct stream_info
void assign_streams(program& p, std::size_t streams) void assign_streams(program& p, std::size_t streams)
{ {
const std::size_t min_partition_threshold = 2; const std::size_t min_partition_threshold = 2;
for(std::size_t stream = 0; stream < streams-1; stream++) for(std::size_t stream = 0; stream < streams - 1; stream++)
{ {
fix([&](auto self, auto ins) { fix([&](auto self, auto ins) {
// If weight is zero then stop // If weight is zero then stop
...@@ -244,7 +244,8 @@ void schedule::apply(program& p) const ...@@ -244,7 +244,8 @@ void schedule::apply(program& p) const
fix([&](auto self, auto ins) { fix([&](auto self, auto ins) {
auto args = ins->inputs(); auto args = ins->inputs();
std::sort(args.begin(), args.end(), [&](auto x, auto y) { std::sort(args.begin(), args.end(), [&](auto x, auto y) {
return std::make_tuple(si.weights[x], x->inputs().size()) < std::make_tuple(si.weights[y], y->inputs().size()); return std::make_tuple(si.weights[x], x->inputs().size()) <
std::make_tuple(si.weights[y], y->inputs().size());
}); });
for(auto i : args) for(auto i : args)
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment