1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
//===-- Unittests for math_extras -----------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "src/__support/UInt128.h" // UInt128
#include "src/__support/integer_literals.h"
#include "src/__support/math_extras.h"
#include "test/UnitTest/Test.h"
namespace LIBC_NAMESPACE {
// TODO: add UInt<128> support.
using UnsignedTypesNoBigInt = testing::TypeList<
#if defined(LIBC_TYPES_HAS_INT128)
__uint128_t,
#endif // LIBC_TYPES_HAS_INT128
unsigned char, unsigned short, unsigned int, unsigned long,
unsigned long long>;
TEST(LlvmLibcBlockMathExtrasTest, mask_trailing_ones) {
EXPECT_EQ(0_u8, (mask_leading_ones<uint8_t, 0>()));
EXPECT_EQ(0_u8, (mask_trailing_ones<uint8_t, 0>()));
EXPECT_EQ(0_u16, (mask_leading_ones<uint16_t, 0>()));
EXPECT_EQ(0_u16, (mask_trailing_ones<uint16_t, 0>()));
EXPECT_EQ(0_u32, (mask_leading_ones<uint32_t, 0>()));
EXPECT_EQ(0_u32, (mask_trailing_ones<uint32_t, 0>()));
EXPECT_EQ(0_u64, (mask_leading_ones<uint64_t, 0>()));
EXPECT_EQ(0_u64, (mask_trailing_ones<uint64_t, 0>()));
EXPECT_EQ(0x00000003_u32, (mask_trailing_ones<uint32_t, 2>()));
EXPECT_EQ(0xC0000000_u32, (mask_leading_ones<uint32_t, 2>()));
EXPECT_EQ(0x000007FF_u32, (mask_trailing_ones<uint32_t, 11>()));
EXPECT_EQ(0xFFE00000_u32, (mask_leading_ones<uint32_t, 11>()));
EXPECT_EQ(0xFFFFFFFF_u32, (mask_trailing_ones<uint32_t, 32>()));
EXPECT_EQ(0xFFFFFFFF_u32, (mask_leading_ones<uint32_t, 32>()));
EXPECT_EQ(0xFFFFFFFFFFFFFFFF_u64, (mask_trailing_ones<uint64_t, 64>()));
EXPECT_EQ(0xFFFFFFFFFFFFFFFF_u64, (mask_leading_ones<uint64_t, 64>()));
EXPECT_EQ(0x0000FFFFFFFFFFFF_u64, (mask_trailing_ones<uint64_t, 48>()));
EXPECT_EQ(0xFFFFFFFFFFFF0000_u64, (mask_leading_ones<uint64_t, 48>()));
EXPECT_EQ(0_u128, (mask_trailing_ones<UInt128, 0>()));
EXPECT_EQ(0_u128, (mask_leading_ones<UInt128, 0>()));
EXPECT_EQ(0x00000000000000007FFFFFFFFFFFFFFF_u128,
(mask_trailing_ones<UInt128, 63>()));
EXPECT_EQ(0xFFFFFFFFFFFFFFFE0000000000000000_u128,
(mask_leading_ones<UInt128, 63>()));
EXPECT_EQ(0x0000000000000000FFFFFFFFFFFFFFFF_u128,
(mask_trailing_ones<UInt128, 64>()));
EXPECT_EQ(0xFFFFFFFFFFFFFFFF0000000000000000_u128,
(mask_leading_ones<UInt128, 64>()));
EXPECT_EQ(0x0000000000000001FFFFFFFFFFFFFFFF_u128,
(mask_trailing_ones<UInt128, 65>()));
EXPECT_EQ(0xFFFFFFFFFFFFFFFF8000000000000000_u128,
(mask_leading_ones<UInt128, 65>()));
EXPECT_EQ(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF_u128,
(mask_trailing_ones<UInt128, 128>()));
EXPECT_EQ(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF_u128,
(mask_leading_ones<UInt128, 128>()));
}
TYPED_TEST(LlvmLibcBitTest, FirstLeadingZero, UnsignedTypesNoBigInt) {
EXPECT_EQ(first_leading_zero<T>(cpp::numeric_limits<T>::max()), 0);
for (int i = 0U; i != cpp::numeric_limits<T>::digits; ++i)
EXPECT_EQ(first_leading_zero<T>(~(T(1) << i)),
cpp::numeric_limits<T>::digits - i);
}
TYPED_TEST(LlvmLibcBitTest, FirstLeadingOne, UnsignedTypesNoBigInt) {
EXPECT_EQ(first_leading_one<T>(static_cast<T>(0)), 0);
for (int i = 0U; i != cpp::numeric_limits<T>::digits; ++i)
EXPECT_EQ(first_leading_one<T>(T(1) << i),
cpp::numeric_limits<T>::digits - i);
}
TYPED_TEST(LlvmLibcBitTest, FirstTrailingZero, UnsignedTypesNoBigInt) {
EXPECT_EQ(first_trailing_zero<T>(cpp::numeric_limits<T>::max()), 0);
for (int i = 0U; i != cpp::numeric_limits<T>::digits; ++i)
EXPECT_EQ(first_trailing_zero<T>(~(T(1) << i)), i + 1);
}
TYPED_TEST(LlvmLibcBitTest, FirstTrailingOne, UnsignedTypesNoBigInt) {
EXPECT_EQ(first_trailing_one<T>(cpp::numeric_limits<T>::max()), 0);
for (int i = 0U; i != cpp::numeric_limits<T>::digits; ++i)
EXPECT_EQ(first_trailing_one<T>(T(1) << i), i + 1);
}
TYPED_TEST(LlvmLibcBitTest, CountZeros, UnsignedTypesNoBigInt) {
EXPECT_EQ(count_zeros(T(0)), cpp::numeric_limits<T>::digits);
for (int i = 0; i != cpp::numeric_limits<T>::digits; ++i)
EXPECT_EQ(count_zeros<T>(cpp::numeric_limits<T>::max() >> i), i);
}
} // namespace LIBC_NAMESPACE
|