Nitload
9647c2c68a
Add: Add factory method to generate more kind of binary operator node Add: more SIMD intrinsic C function Signed-off-by: Nitload <YoumuLiu_NSI@whv0t.onmicrosoft.com>
142 lines
4.6 KiB
C
142 lines
4.6 KiB
C
#include "StgSharpC.h"
|
|
#include "ssc_intrinsic.h"
|
|
|
|
|
|
#ifdef _MSC_VER
|
|
#include <intrin.h>
|
|
#include <isa_availability.h>
|
|
#else
|
|
#include <cpuid.h>
|
|
#endif // _MSC_VER
|
|
|
|
PRIVATE void load_intrinsic_sse(ssc_intrinsic *context);
|
|
PRIVATE void load_intrinsic_avx(ssc_intrinsic *context);
|
|
PRIVATE void load_intrinsic_512(ssc_intrinsic *context);
|
|
|
|
PRIVATE int check_isa(most_advanced_instruction instruction_family)
|
|
{
|
|
if (instruction_family != NEON) {
|
|
#if _MSC_VER
|
|
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
|
|
}
|
|
|
|
SSCAPI int load_intrinsic_function(void *intrinsic_context)
|
|
{
|
|
ssc_null_assert(intrinsic_context);
|
|
|
|
ssc_intrinsic *context = (ssc_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(ssc_intrinsic *context)
|
|
{
|
|
context->add_m2 = addmatrix2_sse;
|
|
context->add_m3 = addmatrix3_sse;
|
|
context->add_m4 = addmatrix4_sse;
|
|
context->transpose_23 = transpose23;
|
|
context->transpose_24 = transpose24;
|
|
context->transpose_32 = transpose32;
|
|
context->transpose_33 = transpose33;
|
|
context->transpose_34 = transpose34;
|
|
context->transpose_42 = transpose42;
|
|
context->transpose_43 = transpose43;
|
|
context->transpose_44 = transpose44;
|
|
context->normalize_3 = normalize;
|
|
context->det_matrix33 = det_mat3;
|
|
context->det_matrix44 = det_mat4;
|
|
context->dot_31 = dot_31_sse;
|
|
context->dot_32 = dot_32_sse;
|
|
context->dot_41 = dot_41_sse;
|
|
context->dot_42 = dot_42_sse;
|
|
context->dot_43 = dot_43_sse;
|
|
context->city_hash_simplify = city_hash_simplify_sse;
|
|
context->index_pair = index_pair_sse;
|
|
context->sub_m2 = submatrix2_sse;
|
|
context->sub_m3 = submatrix3_sse;
|
|
context->sub_m4 = submatrix4_sse;
|
|
|
|
}
|
|
|
|
void load_intrinsic_avx(ssc_intrinsic *context)
|
|
{
|
|
context->add_m2 = addmatrix2_avx;
|
|
context->add_m3 = addmatrix3_avx;
|
|
context->add_m4 = addmatrix4_avx;
|
|
context->transpose_23 = transpose23;
|
|
context->transpose_24 = transpose24;
|
|
context->transpose_32 = transpose32;
|
|
context->transpose_33 = transpose33;
|
|
context->transpose_34 = transpose34;
|
|
context->transpose_42 = transpose42;
|
|
context->transpose_43 = transpose43;
|
|
context->transpose_44 = transpose44;
|
|
context->normalize_3 = normalize;
|
|
context->det_matrix33 = det_mat3;
|
|
context->det_matrix44 = det_mat4;
|
|
context->dot_31 = dot_31_avx;
|
|
context->dot_32 = dot_32_avx;
|
|
context->dot_41 = dot_41_avx;
|
|
context->dot_42 = dot_42_avx;
|
|
context->dot_43 = dot_43_avx;
|
|
context->city_hash_simplify = city_hash_simplify_sse;
|
|
context->index_pair = index_pair_sse;
|
|
}
|
|
|
|
void load_intrinsic_512(ssc_intrinsic *context)
|
|
{
|
|
context->add_m2 = addmatrix2_avx;
|
|
context->add_m3 = addmatrix3_512;
|
|
context->add_m4 = addmatrix4_512;
|
|
context->transpose_23 = transpose23;
|
|
context->transpose_24 = transpose24;
|
|
context->transpose_32 = transpose32;
|
|
context->transpose_33 = transpose33;
|
|
context->transpose_34 = transpose34;
|
|
context->transpose_42 = transpose42;
|
|
context->transpose_43 = transpose43;
|
|
context->transpose_44 = transpose44;
|
|
context->normalize_3 = normalize;
|
|
context->det_matrix33 = det_mat3;
|
|
context->det_matrix44 = det_mat4;
|
|
context->dot_31 = dot_31_avx;
|
|
context->dot_32 = dot_32_avx;
|
|
context->dot_41 = dot_41_avx;
|
|
context->dot_42 = dot_42_avx;
|
|
context->dot_43 = dot_43_avx;
|
|
context->city_hash_simplify = city_hash_simplify_sse;
|
|
context->index_pair = index_pair_sse;
|
|
}
|