Skip to content

GitLab

Commit 863e069b authored by Chao Liu's avatar Chao Liu
Browse files

refactor print array

parent 0ab47e06
Show whitespace changes
Inline Side-by-side
Showing with 179 additions and 177 deletions
composable_kernel/include/utility/print.hpp
View file @ 863e069b
......@@ -12,40 +12,41 @@ __host__ __device__ void print_array(const char* s, T a)
using data_type = typename decltype(a)::data_type;
constexpr index_t nsize = a.Size();
static_assert(nsize > 0 && nsize <= 10, "wrong!");
static_assert(nsize >= 0 && nsize <= 10, "wrong!");
if
constexpr(is_same<data_type, uint32_t>{})
if constexpr(is_same<data_type, uint32_t>{})
{
if constexpr(nsize == 0)
{
printf("%s size %u\n", s, nsize);
}
else if constexpr(nsize == 1)
{
printf("%s size %u, {%u}\n", s, nsize, a[0]);
}
else if constexpr(nsize == 2)
{
printf("%s size %u, {%u %u}\n", s, nsize, a[0], a[1]);
}
else if constexpr(nsize == 3)
{
printf("%s size %u, {%u %u %u}\n", s, nsize, a[0], a[1], a[2]);
}
else if constexpr(nsize == 4)
{
static_if<nsize == 1>{}([&](auto) { printf("%s size %u, {%u}\n", s, nsize, a[0]); });
static_if<nsize == 2>{}(
[&](auto) { printf("%s size %u, {%u %u}\n", s, nsize, a[0], a[1]); });
static_if<nsize == 3>{}(
[&](auto) { printf("%s size %u, {%u %u %u}\n", s, nsize, a[0], a[1], a[2]); });
static_if<nsize == 4>{}([&](auto) {
printf("%s size %u, {%u %u %u %u}\n", s, nsize, a[0], a[1], a[2], a[3]);
});
static_if<nsize == 5>{}([&](auto) {
}
else if constexpr(nsize == 5)
{
printf("%s size %u, {%u %u %u %u %u}\n", s, nsize, a[0], a[1], a[2], a[3], a[4]);
});
static_if<nsize == 6>{}([&](auto) {
printf("%s size %u, {%u %u %u %u %u %u}\n",
s,
nsize,
a[0],
a[1],
a[2],
a[3],
a[4],
a[5]);
});
static_if<nsize == 7>{}([&](auto) {
}
else if constexpr(nsize == 6)
{
printf(
"%s size %u, {%u %u %u %u %u %u}\n", s, nsize, a[0], a[1], a[2], a[3], a[4], a[5]);
}
else if constexpr(nsize == 7)
{
printf("%s size %u, {%u %u %u %u %u %u %u}\n",
s,
nsize,
......@@ -56,9 +57,9 @@ __host__ __device__ void print_array(const char* s, T a)
a[4],
a[5],
a[6]);
});
static_if<nsize == 8>{}([&](auto) {
}
else if constexpr(nsize == 8)
{
printf("%s size %u, {%u %u %u %u %u %u %u %u}\n",
s,
nsize,
......@@ -70,9 +71,9 @@ __host__ __device__ void print_array(const char* s, T a)
a[5],
a[6],
a[7]);
});
static_if<nsize == 9>{}([&](auto) {
}
else if constexpr(nsize == 9)
{
printf("%s size %u, {%u %u %u %u %u %u %u %u %u}\n",
s,
nsize,
......@@ -85,9 +86,9 @@ __host__ __device__ void print_array(const char* s, T a)
a[6],
a[7],
a[8]);
});
static_if<nsize == 10>{}([&](auto) {
}
else if constexpr(nsize == 10)
{
printf("%s size %u, {%u %u %u %u %u %u %u %u %u %u}\n",
s,
nsize,
......@@ -101,40 +102,41 @@ __host__ __device__ void print_array(const char* s, T a)
a[7],
a[8],
a[9]);
});
}
else if
constexpr(is_same<data_type, int32_t>{})
}
else if constexpr(is_same<data_type, int32_t>{})
{
if constexpr(nsize == 0)
{
printf("%s size %d\n", s, nsize);
}
else if constexpr(nsize == 1)
{
printf("%s size %d, {%d}\n", s, nsize, a[0]);
}
else if constexpr(nsize == 2)
{
printf("%s size %d, {%d %d}\n", s, nsize, a[0], a[1]);
}
else if constexpr(nsize == 3)
{
printf("%s size %d, {%d %d %d}\n", s, nsize, a[0], a[1], a[2]);
}
else if constexpr(nsize == 4)
{
static_if<nsize == 1>{}([&](auto) { printf("%s size %d, {%d}\n", s, nsize, a[0]); });
static_if<nsize == 2>{}(
[&](auto) { printf("%s size %d, {%d %d}\n", s, nsize, a[0], a[1]); });
static_if<nsize == 3>{}(
[&](auto) { printf("%s size %d, {%d %d %d}\n", s, nsize, a[0], a[1], a[2]); });
static_if<nsize == 4>{}([&](auto) {
printf("%s size %d, {%d %d %d %d}\n", s, nsize, a[0], a[1], a[2], a[3]);
});
static_if<nsize == 5>{}([&](auto) {
}
else if constexpr(nsize == 5)
{
printf("%s size %d, {%d %d %d %d %d}\n", s, nsize, a[0], a[1], a[2], a[3], a[4]);
});
static_if<nsize == 6>{}([&](auto) {
printf("%s size %d, {%d %d %d %d %d %d}\n",
s,
nsize,
a[0],
a[1],
a[2],
a[3],
a[4],
a[5]);
});
static_if<nsize == 7>{}([&](auto) {
}
else if constexpr(nsize == 6)
{
printf(
"%s size %d, {%d %d %d %d %d %d}\n", s, nsize, a[0], a[1], a[2], a[3], a[4], a[5]);
}
else if constexpr(nsize == 7)
{
printf("%s size %d, {%d %d %d %d %d %d %d}\n",
s,
nsize,
......@@ -145,9 +147,9 @@ __host__ __device__ void print_array(const char* s, T a)
a[4],
a[5],
a[6]);
});
static_if<nsize == 8>{}([&](auto) {
}
else if constexpr(nsize == 8)
{
printf("%s size %d, {%d %d %d %d %d %d %d %d}\n",
s,
nsize,
......@@ -159,9 +161,9 @@ __host__ __device__ void print_array(const char* s, T a)
a[5],
a[6],
a[7]);
});
static_if<nsize == 9>{}([&](auto) {
}
else if constexpr(nsize == 9)
{
printf("%s size %d, {%d %d %d %d %d %d %d %d %d}\n",
s,
nsize,
......@@ -174,9 +176,9 @@ __host__ __device__ void print_array(const char* s, T a)
a[6],
a[7],
a[8]);
});
static_if<nsize == 10>{}([&](auto) {
}
else if constexpr(nsize == 10)
{
printf("%s size %d, {%d %d %d %d %d %d %d %d %d %d}\n",
s,
nsize,
......@@ -190,7 +192,7 @@ __host__ __device__ void print_array(const char* s, T a)
a[7],
a[8],
a[9]);
});
}
}
}
......
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