#include #include #include #include TEST(FP16_IEEE_TO_FP32_PSIMD, infinity) { const uint16_t positive_infinity_f16 = UINT16_C(0x7C00); const uint16_t negative_infinity_f16 = UINT16_C(0xFC00); const uint32_t positive_infinity_f32 = UINT32_C(0x7F800000); const uint32_t negative_infinity_f32 = UINT32_C(0xFF800000); const psimd_u16 fp16 = { positive_infinity_f16, negative_infinity_f16, negative_infinity_f16, positive_infinity_f16 }; const psimd_u32 fp32 = (psimd_u32) fp16_ieee_to_fp32_psimd(fp16); EXPECT_EQ(positive_infinity_f32, fp32[0]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0] << ", " << "F32 = 0x" << std::setw(8) << positive_infinity_f32; EXPECT_EQ(negative_infinity_f32, fp32[1]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1] << ", " << "F32 = 0x" << std::setw(8) << negative_infinity_f32; EXPECT_EQ(negative_infinity_f32, fp32[2]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2] << ", " << "F32 = 0x" << std::setw(8) << negative_infinity_f32; EXPECT_EQ(positive_infinity_f32, fp32[3]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3] << ", " << "F32 = 0x" << std::setw(8) << positive_infinity_f32; } TEST(FP16_IEEE_TO_FP32_PSIMD, positive_nan) { for (uint16_t h = 0; h < 0x0400; h += 4) { const psimd_u16 fp16 = { (uint16_t) (h + 0x7C00 + (h == 0)) /* Avoid infinity */, (uint16_t) (h + 0x7C01), (uint16_t) (h + 0x7C02), (uint16_t) (h + 0x7C03), }; const psimd_u32 fp32 = (psimd_u32) fp16_ieee_to_fp32_psimd(fp16); /* Check sign */ EXPECT_EQ(fp32[0] & UINT32_C(0x80000000), 0) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0]; EXPECT_EQ(fp32[1] & UINT32_C(0x80000000), 0) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1]; EXPECT_EQ(fp32[2] & UINT32_C(0x80000000), 0) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2]; EXPECT_EQ(fp32[3] & UINT32_C(0x80000000), 0) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3]; /* Check exponent */ EXPECT_EQ(fp32[0] & UINT32_C(0x7F800000), UINT32_C(0x7F800000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0]; EXPECT_EQ(fp32[1] & UINT32_C(0x7F800000), UINT32_C(0x7F800000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1]; EXPECT_EQ(fp32[2] & UINT32_C(0x7F800000), UINT32_C(0x7F800000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2]; EXPECT_EQ(fp32[3] & UINT32_C(0x7F800000), UINT32_C(0x7F800000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3]; /* Check mantissa */ EXPECT_NE(fp32[0] & UINT32_C(0x007FFFFF), UINT32_C(0)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0]; EXPECT_NE(fp32[1] & UINT32_C(0x007FFFFF), UINT32_C(0)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1]; EXPECT_NE(fp32[2] & UINT32_C(0x007FFFFF), UINT32_C(0)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2]; EXPECT_NE(fp32[3] & UINT32_C(0x007FFFFF), UINT32_C(0)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3]; } } TEST(FP16_IEEE_TO_FP32_PSIMD, negative_nan) { for (uint16_t h = 0; h < 0x0400; h += 4) { const psimd_u16 fp16 = { (uint16_t) (h + 0xFC00 + (h == 0)) /* Avoid infinity */, (uint16_t) (h + 0xFC01), (uint16_t) (h + 0xFC02), (uint16_t) (h + 0xFC03), }; const psimd_u32 fp32 = (psimd_u32) fp16_ieee_to_fp32_psimd(fp16); /* Check sign */ EXPECT_EQ(fp32[0] & UINT32_C(0x80000000), UINT32_C(0x80000000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0]; EXPECT_EQ(fp32[1] & UINT32_C(0x80000000), UINT32_C(0x80000000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1]; EXPECT_EQ(fp32[2] & UINT32_C(0x80000000), UINT32_C(0x80000000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2]; EXPECT_EQ(fp32[3] & UINT32_C(0x80000000), UINT32_C(0x80000000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3]; /* Check exponent */ EXPECT_EQ(fp32[0] & UINT32_C(0x7F800000), UINT32_C(0x7F800000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0]; EXPECT_EQ(fp32[1] & UINT32_C(0x7F800000), UINT32_C(0x7F800000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1]; EXPECT_EQ(fp32[2] & UINT32_C(0x7F800000), UINT32_C(0x7F800000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2]; EXPECT_EQ(fp32[3] & UINT32_C(0x7F800000), UINT32_C(0x7F800000)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3]; /* Check mantissa */ EXPECT_NE(fp32[0] & UINT32_C(0x007FFFFF), UINT32_C(0)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0]; EXPECT_NE(fp32[1] & UINT32_C(0x007FFFFF), UINT32_C(0)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1]; EXPECT_NE(fp32[2] & UINT32_C(0x007FFFFF), UINT32_C(0)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2]; EXPECT_NE(fp32[3] & UINT32_C(0x007FFFFF), UINT32_C(0)) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3]; } } TEST(FP16_IEEE_TO_FP32_PSIMD, positive_normalized_values) { const uint32_t exponentBias = 15; for (int32_t e = -14; e <= 15; e++) { for (uint16_t h = 0; h < 0x0400; h += 4) { const psimd_u16 fp16 = { (uint16_t) (h + ((e + exponentBias) << 10) + 0), (uint16_t) (h + ((e + exponentBias) << 10) + 1), (uint16_t) (h + ((e + exponentBias) << 10) + 2), (uint16_t) (h + ((e + exponentBias) << 10) + 3), }; const psimd_u32 fp32 = (psimd_u32) fp16_ieee_to_fp32_psimd(fp16); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[0]), fp32[0]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[0]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[1]), fp32[1]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[1]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[2]), fp32[2]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[2]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[3]), fp32[3]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[3]); } } } TEST(FP16_IEEE_TO_FP32_PSIMD, negative_normalized_values) { const uint32_t exponentBias = 15; for (int32_t e = -14; e <= 15; e++) { for (uint16_t h = 0; h < 0x0400; h += 4) { const psimd_u16 fp16 = { (uint16_t) (h + ((e + exponentBias) << 10) + 0x8000), (uint16_t) (h + ((e + exponentBias) << 10) + 0x8001), (uint16_t) (h + ((e + exponentBias) << 10) + 0x8002), (uint16_t) (h + ((e + exponentBias) << 10) + 0x8003), }; const psimd_u32 fp32 = (psimd_u32) fp16_ieee_to_fp32_psimd(fp16); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[0]), fp32[0]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[0]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[1]), fp32[1]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[1]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[2]), fp32[2]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[2]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[3]), fp32[3]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[3]); } } } TEST(FP16_IEEE_TO_FP32_PSIMD, positive_denormalized_values) { for (uint16_t h = 0; h < 0x0400; h += 4) { const psimd_u16 fp16 = { (uint16_t) (h + 0), (uint16_t) (h + 1), (uint16_t) (h + 2), (uint16_t) (h + 3), }; const psimd_u32 fp32 = (psimd_u32) fp16_ieee_to_fp32_psimd(fp16); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[0]), fp32[0]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[0]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[1]), fp32[1]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[1]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[2]), fp32[2]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[2]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[3]), fp32[3]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[3]); } } TEST(FP16_IEEE_TO_FP32_PSIMD, negative_denormalized_values) { for (uint16_t h = 0; h < 0x0400; h += 4) { const psimd_u16 fp16 = { (uint16_t) (h + 0x8000), (uint16_t) (h + 0x8001), (uint16_t) (h + 0x8002), (uint16_t) (h + 0x8003), }; const psimd_u32 fp32 = (psimd_u32) fp16_ieee_to_fp32_psimd(fp16); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[0]), fp32[0]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[0] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[0] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[0]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[1]), fp32[1]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[1] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[1] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[1]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[2]), fp32[2]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[2] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[2] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[2]); EXPECT_EQ(fp16_ieee_to_fp32_bits(fp16[3]), fp32[3]) << std::hex << std::uppercase << std::setfill('0') << "F16 = 0x" << std::setw(4) << fp16[3] << ", " << "F32(F16) = 0x" << std::setw(8) << fp32[3] << ", " << "F32 = 0x" << std::setw(8) << fp16_ieee_to_fp32_bits(fp16[3]); } }