From 08a49953a0954441ceaed836a89a853a597e67f3 Mon Sep 17 00:00:00 2001
From: LiYu Lu <luliyucoordinate@outlook.com>
Date: Fri, 25 Oct 2024 02:39:22 +0800
Subject: [PATCH] Add a print for the uint{x}b_t type. (#1871)

---
 include/cute/util/print.hpp | 62 +++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)

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/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));