Print input streams

05208b60 · Paul · b2b3f07e · 05208b60
Commit 05208b60 authored Mar 05, 2019 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

src/schedule.cpp src/schedule.cpp +6 -0

No files found.
--- a/src/schedule.cpp
+++ b/src/schedule.cpp
@@ -259,6 +259,12 @@ void schedule::apply(program& p) const
        p.annotate(std::cout, [&](auto ins) {
            std::cout << ":";
            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))
                std::cout << " stream=" << si.get_stream(ins);
        });