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

Log more information and initialize with constant value.

parent 406f71a6
...@@ -17,6 +17,18 @@ ...@@ -17,6 +17,18 @@
namespace ck { namespace ck {
namespace test { namespace test {
template <typename Range>
std::string serialize_range(const Range& range)
{
std::stringstream ss;
for(auto& r : range)
{
ss << r << ", ";
}
std::string str = ss.str();
return std::string(str.begin(), str.end() - 2);
}
template <typename Tuple> template <typename Tuple>
class TestGroupedGemm : public testing::TestWithParam<int> class TestGroupedGemm : public testing::TestWithParam<int>
{ {
...@@ -30,7 +42,7 @@ class TestGroupedGemm : public testing::TestWithParam<int> ...@@ -30,7 +42,7 @@ class TestGroupedGemm : public testing::TestWithParam<int>
public: public:
bool verify_ = true; bool verify_ = true;
int init_method_ = 2; // decimal value initialization int init_method_ = 0; // decimal value initialization
bool log_ = false; bool log_ = false;
bool bench_ = false; // measure kernel performance bool bench_ = false; // measure kernel performance
...@@ -44,6 +56,15 @@ class TestGroupedGemm : public testing::TestWithParam<int> ...@@ -44,6 +56,15 @@ class TestGroupedGemm : public testing::TestWithParam<int>
const std::vector<int>& StrideCs, const std::vector<int>& StrideCs,
int kbatch = 1) int kbatch = 1)
{ {
std::cout << "Ms: [" << serialize_range(Ms) << "] "
<< "Ns: [" << serialize_range(Ns) << "] "
<< "Ks: [" << serialize_range(Ks) << "] "
<< "StrideAs: [" << serialize_range(StrideAs) << "] "
<< "StrideBs: [" << serialize_range(StrideBs) << "] "
<< "StrideCs: [" << serialize_range(StrideCs) << "] "
<< "kbatch: " << kbatch << std::endl;
bool pass = ck::profiler::profile_grouped_gemm_impl<ADataType, bool pass = ck::profiler::profile_grouped_gemm_impl<ADataType,
BDataType, BDataType,
EDataType, EDataType,
......
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