llvm/compiler-rt/lib/builtins/aarch64/sme-abi-init.c

// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

__attribute__((visibility("hidden"), nocommon))
_Bool __aarch64_has_sme_and_tpidr2_el0;

// We have multiple ways to check that the function has SME, depending on our
// target.
// * For Linux/Glibc we can use getauxval().
// * For Android we can use getauxval().
// * For newlib we can use __aarch64_sme_accessible().

#if defined(__linux__)

#if defined(__ANDROID__)
#include <sys/auxv.h>
#elif __has_include(<sys/auxv.h>)
#include <sys/auxv.h>
#else
#define getauxval(x) 0
#endif
#include "../cpu_model/aarch64/hwcap.inc"

static _Bool has_sme(void) { return getauxval(AT_HWCAP2) & HWCAP2_SME; }

#else  // defined(__linux__)

#if defined(COMPILER_RT_SHARED_LIB)
__attribute__((weak))
#endif
extern _Bool __aarch64_sme_accessible(void);

static _Bool has_sme(void)  {
#if defined(COMPILER_RT_SHARED_LIB)
  if (!__aarch64_sme_accessible)
    return 0;
#endif
  return __aarch64_sme_accessible();
}

#endif // defined(__linux__)

#if __GNUC__ >= 9
#pragma GCC diagnostic ignored "-Wprio-ctor-dtor"
#endif
__attribute__((constructor(90)))
static void init_aarch64_has_sme(void) {
  __aarch64_has_sme_and_tpidr2_el0 = has_sme();
}