src/vppinfra/cpu.h

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #ifndef included_clib_cpu_h
  17 #define included_clib_cpu_h
  18
  19 #include <vppinfra/format.h>
  20
  21 /*
  22  * multiarchitecture support. Adding new entry will produce
  23  * new graph node function variant optimized for specific cpu
  24  * microarchitecture.
  25  * Order is important for runtime selection, as 1st match wins...
  26  */
  27
  28 #if __x86_64__ && CLIB_DEBUG == 0
  29 #define foreach_march_variant(macro, x) \
  30   macro(avx2,  x, "arch=core-avx2")
  31 #else
  32 #define foreach_march_variant(macro, x)
  33 #endif
  34
  35
  36 #if __GNUC__ > 4  && !__clang__
  37 #define CLIB_CPU_OPTIMIZED __attribute__ ((optimize ("tree-vectorize")))
  38 #else
  39 #define CLIB_CPU_OPTIMIZED
  40 #endif
  41
  42
  43 #define CLIB_MULTIARCH_ARCH_CHECK(arch, fn, tgt)                        \
  44   if (clib_cpu_supports_ ## arch())                                     \
  45     return & fn ## _ ##arch;
  46
  47 #define CLIB_MULTIARCH_SELECT_FN(fn,...)                               \
  48   __VA_ARGS__ void * fn ## _multiarch_select(void)                     \
  49 {                                                                      \
  50   foreach_march_variant(CLIB_MULTIARCH_ARCH_CHECK, fn)                 \
  51   return & fn;                                                         \
  52 }
  53
  54 #ifdef CLIB_MULTIARCH_VARIANT
  55 #define __CLIB_MULTIARCH_FN(a,b) a##_##b
  56 #define _CLIB_MULTIARCH_FN(a,b) __CLIB_MULTIARCH_FN(a,b)
  57 #define CLIB_MULTIARCH_FN(fn) _CLIB_MULTIARCH_FN(fn,CLIB_MULTIARCH_VARIANT)
  58 #else
  59 #define CLIB_MULTIARCH_FN(fn) fn
  60 #endif
  61
  62 #define foreach_x86_64_flags \
  63 _ (sse3,     1, ecx, 0)   \
  64 _ (ssse3,    1, ecx, 9)   \
  65 _ (sse41,    1, ecx, 19)  \
  66 _ (sse42,    1, ecx, 20)  \
  67 _ (avx,      1, ecx, 28)  \
  68 _ (avx2,     7, ebx, 5)   \
  69 _ (avx512f,  7, ebx, 16)  \
  70 _ (aes,      1, ecx, 25)  \
  71 _ (sha,      7, ebx, 29)  \
  72 _ (invariant_tsc, 0x80000007, edx, 8)
  73
  74 #if defined(__x86_64__)
  75 #include "cpuid.h"
  76
  77 static inline int
  78 clib_get_cpuid (const u32 lev, u32 * eax, u32 * ebx, u32 * ecx, u32 * edx)
  79 {
  80   if ((u32) __get_cpuid_max (0x80000000 & lev, 0) < lev)
  81     return 0;
  82   if (lev == 7)
  83     __cpuid_count (lev, 0, *eax, *ebx, *ecx, *edx);
  84   else
  85     __cpuid (lev, *eax, *ebx, *ecx, *edx);
  86   return 1;
  87 }
  88
  89
  90 #define _(flag, func, reg, bit) \
  91 static inline int                                                       \
  92 clib_cpu_supports_ ## flag()                                            \
  93 {                                                                       \
  94   u32 __attribute__((unused)) eax, ebx = 0, ecx = 0, edx  = 0;          \
  95   clib_get_cpuid (func, &eax, &ebx, &ecx, &edx);                        \
  96                                                                         \
  97   return ((reg & (1 << bit)) != 0);                                     \
  98 }
  99 foreach_x86_64_flags
 100 #undef _
 101 #else
 102
 103 #define _(flag, func, reg, bit) \
 104 static inline int clib_cpu_supports_ ## flag() { return 0; }
 105 foreach_x86_64_flags
 106 #undef _
 107 #endif
 108 #endif
 109   format_function_t format_cpu_uarch;
 110 format_function_t format_cpu_model_name;
 111 format_function_t format_cpu_flags;
 112
 113 /*
 114  * fd.io coding-style-patch-verification: ON
 115  *
 116  * Local Variables:
 117  * eval: (c-set-style "gnu")
 118  * End:
 119  */