wanderer/ BLAKE3
1
0
Fork 0
mirror of https://github.com/BLAKE3-team/BLAKE3 synced 2024-05-12 14:46:15 +02:00
Code Issues Releases Activity

Merge 3289c75996 into 4ec3be8bfa

This commit is contained in:
David CARLIER 2024-04-06 12:03:35 +08:00 committed by GitHub
parent 4ec3be8bfa 3289c75996
commit c273deae35
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 28 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
c/blake3_dispatch.c
View File

@ -13,6 +13,12 @@
#else
#undef IS_X86 /* Unimplemented! */
#endif
#else
#if defined(_MSC_VER)
//
#elif defined(__linux__)
#include <sys/auxv.h>
#endif
#endif
#if !defined(BLAKE3_ATOMICS)
@ -88,7 +94,17 @@ static void cpuidex(uint32_t out[4], uint32_t id, uint32_t sid) {
#endif
#if defined(IS_AARCH64)
static int has_neon(void) {
#if defined(__linux__)
return getauxval(AT_HWCAP) & (1 << 1);
#endif
return 1;
}
#endif
enum cpu_feature {
#if defined(IS_X86)
SSE2 = 1 << 0,
SSSE3 = 1 << 1,
SSE41 = 1 << 2,
@ -96,6 +112,9 @@ enum cpu_feature {
AVX2 = 1 << 4,
AVX512F = 1 << 5,
AVX512VL = 1 << 6,
#else
NEON = 1 << 0,
#endif
/* ... */
UNDEFINED = 1 << 30
};
@ -156,8 +175,10 @@ static
ATOMIC_STORE(g_cpu_features, features);
return features;
#else
/* How to detect NEON? */
return 0;
enum cpu_feature features = 0;
if (has_neon())
features |= NEON;
return features;
#endif
}
}
@ -262,9 +283,12 @@ void blake3_hash_many(const uint8_t *const *inputs, size_t num_inputs,
#endif
#if BLAKE3_USE_NEON == 1
blake3_hash_many_neon(inputs, num_inputs, blocks, key, counter,
const enum cpu_feature features = get_cpu_features();
if (features & NEON) {
blake3_hash_many_neon(inputs, num_inputs, blocks, key, counter,
increment_counter, flags, flags_start, flags_end, out);
return;
return;
}
#endif
blake3_hash_many_portable(inputs, num_inputs, blocks, key, counter,