usearch 2.10.2 → 2.10.5

package/fp16/test/peachpy/alt-xmm-to-fp32-ymm-avx2.cc

@@ -0,0 +1,96 @@
+#include <gtest/gtest.h>
+
+#include <cstdint>
+
+#include <fp16.h>
+
+
+extern "C" void fp16_alt_xmm_to_fp32_ymm_peachpy__avx2(const uint16_t* fp16, uint32_t* fp32);
+const size_t vector_elements = 8;
+
+
+TEST(FP16_ALT_XMM_TO_FP32_YMM, positive_normalized_values) {
+	const uint32_t exponentBias = 15;
+	for (int32_t e = -14; e <= 16; e++) {
+		for (uint16_t h = 0; h < 0x0400; h += vector_elements) {
+			uint16_t fp16[vector_elements];
+			for (size_t i = 0; i < vector_elements; i++) {
+				fp16[i] = h + ((e + exponentBias) << 10) + i;
+			}
+			uint32_t fp32[vector_elements];
+			fp16_alt_xmm_to_fp32_ymm_peachpy__avx2(fp16, fp32);
+
+			for (size_t i = 0; i < vector_elements; i++) {
+				EXPECT_EQ(fp16_alt_to_fp32_bits(fp16[i]), fp32[i]) <<
+					std::hex << std::uppercase << std::setfill('0') <<
+					"F16 = 0x" << std::setw(4) << fp16[i] << ", " <<
+					"F32(F16) = 0x" << std::setw(8) << fp32[i] << ", " <<
+					"F32 = 0x" << std::setw(8) << fp16_alt_to_fp32_bits(fp16[i]) <<
+					", lane " << i << "/" << vector_elements;
+			}
+		}
+	}
+}
+
+TEST(FP16_ALT_XMM_TO_FP32_YMM, negative_normalized_values) {
+	const uint32_t exponentBias = 15;
+	for (int32_t e = -14; e <= 16; e++) {
+		for (uint16_t h = 0; h < 0x0400; h += vector_elements) {
+			uint16_t fp16[vector_elements];
+			for (size_t i = 0; i < vector_elements; i++) {
+				fp16[i] = 0x8000 + h + ((e + exponentBias) << 10) + i;
+			}
+			uint32_t fp32[vector_elements];
+			fp16_alt_xmm_to_fp32_ymm_peachpy__avx2(fp16, fp32);
+
+			for (size_t i = 0; i < vector_elements; i++) {
+				EXPECT_EQ(fp16_alt_to_fp32_bits(fp16[i]), fp32[i]) <<
+					std::hex << std::uppercase << std::setfill('0') <<
+					"F16 = 0x" << std::setw(4) << fp16[i] << ", " <<
+					"F32(F16) = 0x" << std::setw(8) << fp32[i] << ", " <<
+					"F32 = 0x" << std::setw(8) << fp16_alt_to_fp32_bits(fp16[i]) <<
+					", lane " << i << "/" << vector_elements;
+			}
+		}
+	}
+}
+
+TEST(FP16_ALT_XMM_TO_FP32_YMM, positive_denormalized_values) {
+	for (uint16_t h = 0; h < 0x0400; h += vector_elements) {
+		uint16_t fp16[vector_elements];
+		for (size_t i = 0; i < vector_elements; i++) {
+			fp16[i] = h + i;
+		}
+		uint32_t fp32[vector_elements];
+		fp16_alt_xmm_to_fp32_ymm_peachpy__avx2(fp16, fp32);
+
+		for (size_t i = 0; i < vector_elements; i++) {
+			EXPECT_EQ(fp16_alt_to_fp32_bits(fp16[i]), fp32[i]) <<
+				std::hex << std::uppercase << std::setfill('0') <<
+				"F16 = 0x" << std::setw(4) << fp16[i] << ", " <<
+				"F32(F16) = 0x" << std::setw(8) << fp32[i] << ", " <<
+				"F32 = 0x" << std::setw(8) << fp16_alt_to_fp32_bits(fp16[i]) <<
+				", lane " << i << "/" << vector_elements;
+		}
+	}
+}
+
+TEST(FP16_ALT_XMM_TO_FP32_YMM, negative_denormalized_values) {
+	for (uint16_t h = 0; h < 0x0400; h += vector_elements) {
+		uint16_t fp16[vector_elements];
+		for (size_t i = 0; i < vector_elements; i++) {
+			fp16[i] = 0x8000 + h + i;
+		}
+		uint32_t fp32[vector_elements];
+		fp16_alt_xmm_to_fp32_ymm_peachpy__avx2(fp16, fp32);
+
+		for (size_t i = 0; i < vector_elements; i++) {
+			EXPECT_EQ(fp16_alt_to_fp32_bits(fp16[i]), fp32[i]) <<
+				std::hex << std::uppercase << std::setfill('0') <<
+				"F16 = 0x" << std::setw(4) << fp16[i] << ", " <<
+				"F32(F16) = 0x" << std::setw(8) << fp32[i] << ", " <<
+				"F32 = 0x" << std::setw(8) << fp16_alt_to_fp32_bits(fp16[i]) <<
+				", lane " << i << "/" << vector_elements;
+		}
+	}
+}

package/fp16/test/peachpy/stubs.py

@@ -0,0 +1,38 @@
+from peachpy import *
+from peachpy.x86_64 import *
+
+import fp16.avx, fp16.avx2
+
+
+arg_fp16 = Argument(ptr(const_uint16_t), name="fp16")
+arg_fp32 = Argument(ptr(uint32_t), name="fp32")
+
+with Function("fp16_alt_xmm_to_fp32_ymm_peachpy__avx2", (arg_fp16, arg_fp32), target=uarch.default + isa.avx2):
+
+    reg_fp16 = GeneralPurposeRegister64()
+    LOAD.ARGUMENT(reg_fp16, arg_fp16)
+
+    reg_fp32 = GeneralPurposeRegister64()
+    LOAD.ARGUMENT(reg_fp32, arg_fp32)
+
+    xmm_fp16 = XMMRegister()
+    VMOVUPS(xmm_fp16, [reg_fp16])
+    ymm_fp32 = fp16.avx2.fp16_alt_xmm_to_fp32_ymm(xmm_fp16)
+    VMOVUPS([reg_fp32], ymm_fp32)
+
+    RETURN()
+
+with Function("fp16_alt_xmm_to_fp32_xmm_peachpy__avx", (arg_fp16, arg_fp32), target=uarch.default + isa.avx):
+
+    reg_fp16 = GeneralPurposeRegister64()
+    LOAD.ARGUMENT(reg_fp16, arg_fp16)
+
+    reg_fp32 = GeneralPurposeRegister64()
+    LOAD.ARGUMENT(reg_fp32, arg_fp32)
+
+    xmm_fp16 = XMMRegister()
+    VMOVUPS(xmm_fp16, [reg_fp16])
+    xmm_fp32 = fp16.avx.fp16_alt_xmm_to_fp32_xmm(xmm_fp16)
+    VMOVUPS([reg_fp32], xmm_fp32)
+
+    RETURN()