diff --git a/include/cute/util/print.hpp b/include/cute/util/print.hpp index 6bfe6c0a..dbd65816 100644 --- a/include/cute/util/print.hpp +++ b/include/cute/util/print.hpp @@ -32,6 +32,7 @@ #include // CUTE_HOST_DEVICE #include // cute::is_valid +#include // // CUDA compatible print and printf @@ -96,6 +97,42 @@ print(int a) { printf("%d", a); } +CUTE_HOST_DEVICE +void +print(uint1b_t a) { + printf("%d", int(a)); +} + +CUTE_HOST_DEVICE +void +print(int2b_t a) { + printf("%d", int(a)); +} + +CUTE_HOST_DEVICE +void +print(uint2b_t a) { + printf("%d", int(a)); +} + +CUTE_HOST_DEVICE +void +print(int4b_t a) { + printf("%d", int(a)); +} + +CUTE_HOST_DEVICE +void +print(uint4b_t a) { + printf("%d", int(a)); +} + +CUTE_HOST_DEVICE +void +print(bin1_t a) { + printf("%d", int(a)); +} + CUTE_HOST_DEVICE void print(unsigned int a) { @@ -155,6 +192,31 @@ print(char const* format) { // pretty printing // +CUTE_HOST_DEVICE void +pretty_print(uint1b_t a) { + printf("%*d", 3, int(a)); +} + +CUTE_HOST_DEVICE void +pretty_print(int2b_t a) { + printf("%*d", 5, int(a)); +} + +CUTE_HOST_DEVICE void +pretty_print(uint2b_t a) { + printf("%*d", 5, int(a)); +} + +CUTE_HOST_DEVICE void +pretty_print(int4b_t a) { + printf("%*d", 5, int(a)); +} + +CUTE_HOST_DEVICE void +pretty_print(uint4b_t a) { + printf("%*d", 5, int(a)); +} + CUTE_HOST_DEVICE void pretty_print(bool v) { printf("%*d", 3, int(v));