diff --git a/include/cute/util/print_tensor.hpp b/include/cute/util/print_tensor.hpp index c5eb39a1d5..526fa63d3f 100644 --- a/include/cute/util/print_tensor.hpp +++ b/include/cute/util/print_tensor.hpp @@ -50,7 +50,6 @@ print_layout(Layout const& layout) // (m,n) -> idx CUTE_STATIC_ASSERT_V(rank(layout) == Int<2>{}); int idx_width = num_digits(cosize(layout)) + 2; - const char* delim = "+-----------------------"; print(layout); print("\n"); @@ -63,7 +62,12 @@ print_layout(Layout const& layout) // (m,n) -> idx for (int m = 0; m < size<0>(layout); ++m) { // Header print(" "); - for (int n = 0; n < size<1>(layout); ++n) { printf("%.*s", idx_width+1, delim); } + for (int n = 0; n < size<1>(layout); ++n) { + printf("+"); + for (int i = 0; i < idx_width; ++i) { + printf("-"); + } + } printf("+\n"); // Values printf("%2d ", m); // Row indices @@ -72,7 +76,12 @@ print_layout(Layout const& layout) // (m,n) -> idx } // Footer print(" "); - for (int n = 0; n < size<1>(layout); ++n) { printf("%.*s", idx_width+1, delim); } + for (int n = 0; n < size<1>(layout); ++n) { + printf("+"); + for (int i = 0; i < idx_width; ++i) { + printf("-"); + } + } printf("+\n"); }