Commit d452452d authored by Adam Osewski's avatar Adam Osewski
Browse files

Add stream operator overload for Sequence data type.

parent 920a752b
...@@ -9,6 +9,8 @@ ...@@ -9,6 +9,8 @@
#include <iterator> #include <iterator>
#include "ck/tensor_description/tensor_descriptor.hpp" #include "ck/tensor_description/tensor_descriptor.hpp"
#include "ck/utility/sequence.hpp"
#include "ck/utility/functional2.hpp"
template <typename T> template <typename T>
std::ostream& operator<<(std::ostream& os, const std::vector<T>& v) std::ostream& operator<<(std::ostream& os, const std::vector<T>& v)
...@@ -39,3 +41,13 @@ std::ostream& operator<<(std::ostream& os, const ck::TensorDescriptor<Ts...>& de ...@@ -39,3 +41,13 @@ std::ostream& operator<<(std::ostream& os, const ck::TensorDescriptor<Ts...>& de
return os; return os;
} }
template <ck::index_t... Is>
std::ostream& operator<<(std::ostream& os, const ck::Sequence<Is...>& seq)
{
os << "{";
ck::static_for<0, seq.Size(), 1>{}([&](auto i) { os << seq.At(i).value << ", "; });
os << "}";
return os;
}
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