aboutsummaryrefslogtreecommitdiff
path: root/sha3.c
diff options
context:
space:
mode:
authorPaul Duncan <pabs@pablotron.org>2024-05-05 11:41:46 -0400
committerPaul Duncan <pabs@pablotron.org>2024-05-05 11:41:46 -0400
commit6678b734ef7e91e0cf91d748910b778891a53e88 (patch)
treecd94781cb48f54d31e45a59337512d2263617245 /sha3.c
parentc8536f5094f58db837e5d30f5c24aa1aaabe8eef (diff)
downloadsha3-6678b734ef7e91e0cf91d748910b778891a53e88.tar.bz2
sha3-6678b734ef7e91e0cf91d748910b778891a53e88.zip
sha3.c, Makefile, tests/bench/Makefile: allow overriding SHA3_BACKEND via command-line argument
Diffstat (limited to 'sha3.c')
-rw-r--r--sha3.c20
1 files changed, 14 insertions, 6 deletions
diff --git a/sha3.c b/sha3.c
index 329a810..4073287 100644
--- a/sha3.c
+++ b/sha3.c
@@ -26,20 +26,28 @@
/** @cond INTERNAL */
// available backends
-#define BACKEND_AVX512 8 // AVX-512 backend
-#define BACKEND_NEON 4 // A64 Neon backend
-#define BACKEND_SCALAR 0 // scalar (default) backend
-
-// auto-detect backend
+#define BACKEND_AUTO 0 // auto-detect (default)
+#define BACKEND_SCALAR 1 // scalar backend
+#define BACKEND_AVX512 2 // AVX-512 backend
+#define BACKEND_NEON 3 // Neon backend
+
+// if SHA3_BACKEND is defined and set to 0 (the default), then unset it
+// and auto-detect the appropriate backend
+#if defined(SHA3_BACKEND) && SHA3_BACKEND == BACKEND_AUTO
+#undef SHA3_BACKEND
+#endif /* defined(SHA3_BACKEND) && SHA3_BACKEND == 0 */
+
+// detect backend
#ifndef SHA3_BACKEND
#if defined(__AVX512F__)
#define SHA3_BACKEND BACKEND_AVX512
#elif defined(__ARM_NEON)
#define SHA3_BACKEND BACKEND_NEON
#else
+// no optimized backend detected, fall back to scalar
#define SHA3_BACKEND BACKEND_SCALAR
#endif
-#endif /* SHA3_BACKEND */
+#endif /* !SHA3_BACKEND */
// 64-bit rotate left
#define ROL(v, n) (((v) << (n)) | ((v) >> (64-(n))))