arm_cpudetect.c 3.89 KB
Newer Older
1
/*
Yaowu Xu's avatar
Yaowu Xu committed
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3
 *
Yaowu Xu's avatar
Yaowu Xu committed
4 5 6 7 8 9
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10 11 12 13
 */

#include <stdlib.h>
#include <string.h>
14
#include "aom_ports/arm.h"
Yaowu Xu's avatar
Yaowu Xu committed
15
#include "./aom_config.h"
16

17 18 19 20 21 22 23
#ifdef WINAPI_FAMILY
#include <winapifamily.h>
#if !WINAPI_FAMILY_PARTITION(WINAPI_PARTITION_DESKTOP)
#define getenv(x) NULL
#endif
#endif

John Koleszar's avatar
John Koleszar committed
24 25
static int arm_cpu_env_flags(int *flags) {
  char *env;
Yaowu Xu's avatar
Yaowu Xu committed
26
  env = getenv("AOM_SIMD_CAPS");
John Koleszar's avatar
John Koleszar committed
27 28 29 30 31 32
  if (env && *env) {
    *flags = (int)strtol(env, NULL, 0);
    return 0;
  }
  *flags = 0;
  return -1;
33 34
}

John Koleszar's avatar
John Koleszar committed
35 36
static int arm_cpu_env_mask(void) {
  char *env;
Yaowu Xu's avatar
Yaowu Xu committed
37
  env = getenv("AOM_SIMD_CAPS_MASK");
John Koleszar's avatar
John Koleszar committed
38
  return env && *env ? (int)strtol(env, NULL, 0) : ~0;
39 40
}

John Koleszar's avatar
John Koleszar committed
41
#if !CONFIG_RUNTIME_CPU_DETECT
42

John Koleszar's avatar
John Koleszar committed
43 44 45 46 47 48 49 50 51 52
int arm_cpu_caps(void) {
  /* This function should actually be a no-op. There is no way to adjust any of
   * these because the RTCD tables do not exist: the functions are called
   * statically */
  int flags;
  int mask;
  if (!arm_cpu_env_flags(&flags)) {
    return flags;
  }
  mask = arm_cpu_env_mask();
53
#if HAVE_NEON || HAVE_NEON_ASM
John Koleszar's avatar
John Koleszar committed
54
  flags |= HAS_NEON;
55
#endif /* HAVE_NEON  || HAVE_NEON_ASM */
John Koleszar's avatar
John Koleszar committed
56 57 58 59
  return flags & mask;
}

#elif defined(_MSC_VER) /* end !CONFIG_RUNTIME_CPU_DETECT */
60 61 62 63 64
/*For GetExceptionCode() and EXCEPTION_ILLEGAL_INSTRUCTION.*/
#define WIN32_LEAN_AND_MEAN
#define WIN32_EXTRA_LEAN
#include <windows.h>

John Koleszar's avatar
John Koleszar committed
65 66 67 68 69 70 71
int arm_cpu_caps(void) {
  int flags;
  int mask;
  if (!arm_cpu_env_flags(&flags)) {
    return flags;
  }
  mask = arm_cpu_env_mask();
clang-format's avatar
clang-format committed
72 73 74 75
/* MSVC has no inline __asm support for ARM, but it does let you __emit
 *  instructions via their assembled hex code.
 * All of these instructions should be essentially nops.
 */
76
#if HAVE_NEON || HAVE_NEON_ASM
clang-format's avatar
clang-format committed
77 78 79 80 81 82 83 84
  if (mask & HAS_NEON) {
    __try {
      /*VORR q0,q0,q0*/
      __emit(0xF2200150);
      flags |= HAS_NEON;
    } __except (GetExceptionCode() == EXCEPTION_ILLEGAL_INSTRUCTION) {
      /*Ignore exception.*/
    }
John Koleszar's avatar
John Koleszar committed
85
  }
86
#endif /* HAVE_NEON || HAVE_NEON_ASM */
clang-format's avatar
clang-format committed
87
  return flags & mask;
88 89
}

John Koleszar's avatar
John Koleszar committed
90 91 92 93 94 95 96 97 98 99 100 101 102
#elif defined(__ANDROID__) /* end _MSC_VER */
#include <cpu-features.h>

int arm_cpu_caps(void) {
  int flags;
  int mask;
  uint64_t features;
  if (!arm_cpu_env_flags(&flags)) {
    return flags;
  }
  mask = arm_cpu_env_mask();
  features = android_getCpuFeatures();

103
#if HAVE_NEON || HAVE_NEON_ASM
clang-format's avatar
clang-format committed
104
  if (features & ANDROID_CPU_ARM_FEATURE_NEON) flags |= HAS_NEON;
105
#endif /* HAVE_NEON || HAVE_NEON_ASM */
John Koleszar's avatar
John Koleszar committed
106 107 108
  return flags & mask;
}

109 110
#elif defined(__linux__) /* end __ANDROID__ */

111 112
#include <stdio.h>

John Koleszar's avatar
John Koleszar committed
113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
int arm_cpu_caps(void) {
  FILE *fin;
  int flags;
  int mask;
  if (!arm_cpu_env_flags(&flags)) {
    return flags;
  }
  mask = arm_cpu_env_mask();
  /* Reading /proc/self/auxv would be easier, but that doesn't work reliably
   *  on Android.
   * This also means that detection will fail in Scratchbox.
   */
  fin = fopen("/proc/cpuinfo", "r");
  if (fin != NULL) {
    /* 512 should be enough for anybody (it's even enough for all the flags
     * that x86 has accumulated... so far).
129
     */
John Koleszar's avatar
John Koleszar committed
130 131
    char buf[512];
    while (fgets(buf, 511, fin) != NULL) {
Johann's avatar
Johann committed
132
#if HAVE_NEON || HAVE_NEON_ASM
John Koleszar's avatar
John Koleszar committed
133 134 135 136 137 138 139
      if (memcmp(buf, "Features", 8) == 0) {
        char *p;
        p = strstr(buf, " neon");
        if (p != NULL && (p[5] == ' ' || p[5] == '\n')) {
          flags |= HAS_NEON;
        }
      }
Johann's avatar
Johann committed
140
#endif /* HAVE_NEON || HAVE_NEON_ASM */
141
    }
John Koleszar's avatar
John Koleszar committed
142 143 144
    fclose(fin);
  }
  return flags & mask;
145
}
clang-format's avatar
clang-format committed
146 147 148
#else  /* end __linux__ */
#error \
    "--enable-runtime-cpu-detect selected, but no CPU detection method " \
John Koleszar's avatar
John Koleszar committed
149
"available for your platform. Reconfigure with --disable-runtime-cpu-detect."
150
#endif