youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
LiYu Lu
2024-10-25 02:39:22 +08:00
committed by GitHub
parent a424ca6cf9
commit 08a49953a0
1 changed files with 62 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
include/cute/util/print.hpp
View File

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