1 | #include "benchmarks/gpu/BenchmarkLogger.h" |
2 | #include "src/__support/CPP/string.h" |
3 | #include "src/__support/CPP/string_view.h" |
4 | #include "src/__support/OSUtil/io.h" // write_to_stderr |
5 | #include "src/__support/big_int.h" // is_big_int |
6 | #include "src/__support/macros/config.h" |
7 | #include "src/__support/macros/properties/types.h" // LIBC_TYPES_HAS_INT128 |
8 | #include "src/__support/uint128.h" |
9 | |
10 | #include <stdint.h> |
11 | |
12 | namespace LIBC_NAMESPACE_DECL { |
13 | namespace benchmarks { |
14 | |
15 | // cpp::string_view specialization |
16 | template <> |
17 | BenchmarkLogger & |
18 | BenchmarkLogger::operator<< <cpp::string_view>(cpp::string_view str) { |
19 | LIBC_NAMESPACE::write_to_stderr(str); |
20 | return *this; |
21 | } |
22 | |
23 | // cpp::string specialization |
24 | template <> |
25 | BenchmarkLogger &BenchmarkLogger::operator<< <cpp::string>(cpp::string str) { |
26 | return *this << static_cast<cpp::string_view>(str); |
27 | } |
28 | |
29 | // const char* specialization |
30 | template <> |
31 | BenchmarkLogger &BenchmarkLogger::operator<< <const char *>(const char *str) { |
32 | return *this << cpp::string_view(str); |
33 | } |
34 | |
35 | // char* specialization |
36 | template <> BenchmarkLogger &BenchmarkLogger::operator<< <char *>(char *str) { |
37 | return *this << cpp::string_view(str); |
38 | } |
39 | |
40 | // char specialization |
41 | template <> BenchmarkLogger &BenchmarkLogger::operator<<(char ch) { |
42 | return *this << cpp::string_view(&ch, 1); |
43 | } |
44 | |
45 | // bool specialization |
46 | template <> BenchmarkLogger &BenchmarkLogger::operator<<(bool cond) { |
47 | return *this << (cond ? "true" : "false" ); |
48 | } |
49 | |
50 | // void * specialization |
51 | template <> BenchmarkLogger &BenchmarkLogger::operator<<(void *addr) { |
52 | return *this << "0x" << cpp::to_string(reinterpret_cast<uintptr_t>(addr)); |
53 | } |
54 | |
55 | template <typename T> BenchmarkLogger &BenchmarkLogger::operator<<(T t) { |
56 | if constexpr (is_big_int_v<T> || |
57 | (cpp::is_integral_v<T> && cpp::is_unsigned_v<T> && |
58 | (sizeof(T) > sizeof(uint64_t)))) { |
59 | static_assert(sizeof(T) % 8 == 0, "Unsupported size of UInt" ); |
60 | const IntegerToString<T, radix::Hex::WithPrefix> buffer(t); |
61 | return *this << buffer.view(); |
62 | } else { |
63 | return *this << cpp::to_string(t); |
64 | } |
65 | } |
66 | |
67 | // is_integral specializations |
68 | // char is already specialized to handle character |
69 | template BenchmarkLogger &BenchmarkLogger::operator<< <short>(short); |
70 | template BenchmarkLogger &BenchmarkLogger::operator<< <int>(int); |
71 | template BenchmarkLogger &BenchmarkLogger::operator<< <long>(long); |
72 | template BenchmarkLogger &BenchmarkLogger::operator<< <long long>(long long); |
73 | template BenchmarkLogger & |
74 | BenchmarkLogger::operator<< <unsigned char>(unsigned char); |
75 | template BenchmarkLogger & |
76 | BenchmarkLogger::operator<< <unsigned short>(unsigned short); |
77 | template BenchmarkLogger & |
78 | BenchmarkLogger::operator<< <unsigned int>(unsigned int); |
79 | template BenchmarkLogger & |
80 | BenchmarkLogger::operator<< <unsigned long>(unsigned long); |
81 | template BenchmarkLogger & |
82 | BenchmarkLogger::operator<< <unsigned long long>(unsigned long long); |
83 | |
84 | #ifdef LIBC_TYPES_HAS_INT128 |
85 | template BenchmarkLogger & |
86 | BenchmarkLogger::operator<< <__uint128_t>(__uint128_t); |
87 | #endif // LIBC_TYPES_HAS_INT128 |
88 | template BenchmarkLogger &BenchmarkLogger::operator<< <UInt<128>>(UInt<128>); |
89 | template BenchmarkLogger &BenchmarkLogger::operator<< <UInt<192>>(UInt<192>); |
90 | template BenchmarkLogger &BenchmarkLogger::operator<< <UInt<256>>(UInt<256>); |
91 | template BenchmarkLogger &BenchmarkLogger::operator<< <UInt<320>>(UInt<320>); |
92 | |
93 | // TODO: Add floating point formatting once it's supported by StringStream. |
94 | |
95 | BenchmarkLogger log; |
96 | |
97 | } // namespace benchmarks |
98 | } // namespace LIBC_NAMESPACE_DECL |
99 | |