Commit 05208b60 authored by Paul's avatar Paul
Browse files

Print input streams

parent b2b3f07e
...@@ -259,6 +259,12 @@ void schedule::apply(program& p) const ...@@ -259,6 +259,12 @@ void schedule::apply(program& p) const
p.annotate(std::cout, [&](auto ins) { p.annotate(std::cout, [&](auto ins) {
std::cout << ":"; std::cout << ":";
std::cout << " weight=" << si.weights.at(ins); std::cout << " weight=" << si.weights.at(ins);
std::cout << " input={";
si.get_streams(ins, get_inputs())([&](auto s) {
std::cout << s << ",";
return true;
});
std::cout << "}";
if(si.has_stream(ins)) if(si.has_stream(ins))
std::cout << " stream=" << si.get_stream(ins); std::cout << " stream=" << si.get_stream(ins);
}); });
......
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