Add a print for the uint{x}b_t type. (#1871)

NVIDIA · Oct 24, 2024 · 08a4995 · 08a4995
1 parent a424ca6
commit 08a4995
Showing 1 changed file with 62 additions and 0 deletions.
diff --git a/include/cute/util/print.hpp b/include/cute/util/print.hpp
@@ -32,6 +32,7 @@
 
 #include <cute/config.hpp>           // CUTE_HOST_DEVICE
 #include <cute/util/type_traits.hpp> // cute::is_valid
+#include <cute/numeric/numeric_types.hpp> 
 
 //
 // 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));