add missed files

2025-08-31 14:34:44 +08:00
parent 6bebe1ca85
commit b650526985
21 changed files with 2031 additions and 0 deletions
@@ -0,0 +1,101 @@
+#include "StgSharpNative.h"
+#include "sn_intrinsic.h"
+
+
+#if defined(_MSC_VER) && !defined(__clang__)
+#include <intrin.h>
+#include <isa_availability.h>
+#else
+#include <cpuid.h>
+#endif //  _MSC_VER
+
+PRIVATE void load_intrinsic_sse(sn_intrinsic *context);
+PRIVATE void load_intrinsic_avx(sn_intrinsic *context);
+PRIVATE void load_intrinsic_512(sn_intrinsic *context);
+
+PRIVATE int check_isa(most_advanced_instruction instruction_family)
+{
+        if (instruction_family != NEON) {
+#if defined(_MSC_VER) && !defined(__clang__)
+                return __check_isa_support(instruction_family, 0);
+#else
+                switch (instruction_family) {
+                case SSE:
+                        return __builtin_cpu_supports("sse4.2");
+                case AVX256:
+                        return __builtin_cpu_supports("avx2");
+                case AVX512:
+                        return __builtin_cpu_supports("avx512f");
+                default:
+                        return 0;
+                }
+#endif //  _MSC_VER
+        }
+
+#if defined(__ARM_NEON) || defined(__ARM_NEON__)
+        return 1;
+#else
+        return 0;
+#endif
+}
+
+SN_API int load_intrinsic_function(void *intrinsic_context)
+{
+        sn_null_assert(intrinsic_context);
+
+        sn_intrinsic *context = (sn_intrinsic *)intrinsic_context;
+
+        if (check_isa(AVX512)) {
+                load_intrinsic_512(context);
+                return 3;
+        } else if (check_isa(AVX256)) {
+                load_intrinsic_avx(context);
+                return 2;
+        } else if (check_isa(SSE)) {
+                load_intrinsic_sse(context);
+                return 1;
+        } else if (check_isa(NEON)) {
+                return -1;
+        }
+
+        return 0;
+}
+
+void load_intrinsic_sse(sn_intrinsic *context)
+{
+        context->city_hash_simplify = city_hash_simplify_sse;
+        context->factorial_simd = factorial_simd_sse;
+        context->index_pair = index_pair_sse;
+        context->kernel_add = (MATARITHMETIC)kernel_add_sse;
+        context->kernel_fma = (void(*)(void*, void*, void*))kernel_fma_sse;
+        context->kernel_scalar_mul = kernel_scalar_mul_sse;
+        context->kernel_sub = (MATARITHMETIC)kernel_sub_sse;
+        context->kernel_transpose = (MATTRANSPOSEPROC)kernel_transpose_sse;
+        context->normalize_3 = normalize;
+}
+
+void load_intrinsic_avx(sn_intrinsic *context)
+{
+        context->city_hash_simplify = city_hash_simplify_sse;
+        context->factorial_simd = factorial_simd_sse;
+        context->index_pair = index_pair_sse;
+        context->kernel_add = (MATARITHMETIC)kernel_add_avx;
+        context->kernel_fma = (void(*)(void*, void*, void*))kernel_fma_sse;
+        context->kernel_scalar_mul = kernel_scalar_mul_avx;
+        context->kernel_sub = (MATARITHMETIC)kernel_sub_avx;
+        context->kernel_transpose = (MATTRANSPOSEPROC)kernel_transpose_avx;
+        context->normalize_3 = normalize;
+}
+
+void load_intrinsic_512(sn_intrinsic *context)
+{
+        context->city_hash_simplify = city_hash_simplify_sse;
+        context->factorial_simd = factorial_simd_sse;
+        context->index_pair = index_pair_sse;
+        context->kernel_add = (MATARITHMETIC)kernel_add_512;
+        context->kernel_fma = (void(*)(void*, void*, void*))kernel_fma_sse;
+        context->kernel_scalar_mul = kernel_scalar_mul_512;
+        context->kernel_sub = (MATARITHMETIC)kernel_sub_512;
+        context->kernel_transpose = (MATTRANSPOSEPROC)kernel_transpose_512;
+        context->normalize_3 = normalize;
+}