libc/test/src/__support/FPUtil/bfloat16_test.cpp


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

//===-- Unit tests for bfloat16 type --------------------------------------===//
//
// 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/FPUtil/bfloat16.h"
#include "test/UnitTest/FPMatcher.h"
#include "test/UnitTest/Test.h"
#include "utils/MPFRWrapper/MPCommon.h"

using BFloat16 = LIBC_NAMESPACE::fputil::BFloat16;
using LlvmLibcBfloat16ConversionTest =
    LIBC_NAMESPACE::testing::FPTest<BFloat16>;

// range: [0, inf]
static constexpr uint16_t POS_START = 0x0000U;
static constexpr uint16_t POS_STOP = 0x7f80U;

// range: [-0, -inf]
static constexpr uint16_t NEG_START = 0x8000U;
static constexpr uint16_t NEG_STOP = 0xff80U;

using MPFRNumber = LIBC_NAMESPACE::testing::mpfr::MPFRNumber;

TEST_F(LlvmLibcBfloat16ConversionTest, ToFloatPositiveRange) {
  for (uint16_t bits = POS_START; bits <= POS_STOP; bits++) {
    BFloat16 bf16_num{bits};
    MPFRNumber mpfr_num{bf16_num};

    // bfloat16 to float
    float mpfr_float = mpfr_num.as<float>();
    EXPECT_FP_EQ_ALL_ROUNDING(mpfr_float, static_cast<float>(bf16_num));

    // float to bfloat16
    BFloat16 bf16_from_float{mpfr_float};
    MPFRNumber mpfr_num_2{mpfr_float};
    BFloat16 mpfr_bfloat = mpfr_num_2.as<BFloat16>();
    EXPECT_FP_EQ_ALL_ROUNDING(mpfr_bfloat, bf16_from_float);
  }
}

TEST_F(LlvmLibcBfloat16ConversionTest, ToFloatNegativeRange) {
  for (uint16_t bits = NEG_START; bits <= NEG_STOP; bits++) {
    BFloat16 bf16_num{bits};
    MPFRNumber mpfr_num{bf16_num};

    // bfloat16 to float
    float mpfr_float = mpfr_num.as<float>();
    EXPECT_FP_EQ_ALL_ROUNDING(mpfr_float, static_cast<float>(bf16_num));

    // float to bfloat16
    BFloat16 bf16_from_float{mpfr_float};
    MPFRNumber mpfr_num_2{mpfr_float};
    BFloat16 mpfr_bfloat = mpfr_num_2.as<BFloat16>();
    EXPECT_FP_EQ_ALL_ROUNDING(mpfr_bfloat, bf16_from_float);
  }
}

TEST_F(LlvmLibcBfloat16ConversionTest, FromInteger) {
  constexpr int RANGE = 100'000;
  for (int i = -RANGE; i <= RANGE; i++) {
    BFloat16 mpfr_bfloat = MPFRNumber(i).as<BFloat16>();
    BFloat16 libc_bfloat{i};
    EXPECT_FP_EQ_ALL_ROUNDING(mpfr_bfloat, libc_bfloat);
  }
}