From 4616467c4e68a3fbe7e59bee04863bcc7c5475f1 Mon Sep 17 00:00:00 2001 From: Anthony Scemama Date: Tue, 17 Jan 2023 15:40:00 +0100 Subject: [PATCH] Added SIMD_LENGTH detection --- configure.ac | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/configure.ac b/configure.ac index 7605ac4..6be7f11 100644 --- a/configure.ac +++ b/configure.ac @@ -75,6 +75,7 @@ AS_IF([test "x$with_icc.$with_ifort" = "xyes.yes"], [ LAPACK_LIBS=""]) # Default Automake variables +F77=$FC AM_PROG_AR AM_MAINTAINER_MODE() LT_INIT @@ -213,6 +214,32 @@ AS_CASE([$with_trexio], PKG_CFLAGS2="$PKG_CFLAGS2 $TREXIO_CFLAGS" PKG_LIBS2="$PKG_LIBS2 $TREXIO_LIBS" + +# Checking SIMD +AC_MSG_CHECKING([for SIMD length]) +AC_RUN_IFELSE( + [AC_LANG_PROGRAM([],[ +int simd=1; +#if defined(__AVX512F__) + simd=8; +#elif defined(__AVX__) + simd=4; +#elif defined(__SSE2__) + simd=2; +#elif defined(__ARM_NEON__) + simd=2; +#endif + return simd; +])], +[ SIMD_LENGTH=1 ], +[ SIMD_LENGTH=$? ], +[ SIMD_LENGTH=1 ] +) +AC_MSG_RESULT([$SIMD_LENGTH]) +AC_DEFINE_UNQUOTED([SIMD_LENGTH], [$SIMD_LENGTH], [Length of SIMD vectors]) + + + # QMCKLDGEMM AC_ARG_WITH([qmckldgemm], [AS_HELP_STRING([--with-qmckldgemm],