Print out info about convolution

fd6f1cc6 · Paul · 2b48efe7 · fd6f1cc6
Commit fd6f1cc6 authored Sep 18, 2018 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

src/targets/gpu/lowering.cpp src/targets/gpu/lowering.cpp +9 -0

No files found.
--- a/src/targets/gpu/lowering.cpp
+++ b/src/targets/gpu/lowering.cpp
@@ -131,6 +131,15 @@ struct miopen_convolution
        algo = perf.fwd_algo;
        return shape{shape::int8_type, {perf.memory}};
    }
+    friend std::ostream& operator<<(std::ostream& os, const miopen_convolution& self)
+    {
+        os << self.name() << "[";
+        os << self.op << ", ";
+        os << "algo=" << self.algo;
+        os << "]";
+        return os;
+    }
 };
 struct miopen_pooling