1 /* 2 * Copyright 2011 The LibYuv Project Authors. All rights reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11 #ifndef INCLUDE_LIBYUV_CPU_ID_H_ // NOLINT 12 #define INCLUDE_LIBYUV_CPU_ID_H_ 13 14 #include "libyuv/basic_types.h" 15 16 #ifdef __cplusplus 17 namespace libyuv { 18 extern "C" { 19 #endif 20 21 // Internal flag to indicate cpuid requires initialization. 22 static const int kCpuInitialized = 0x1; 23 24 // These flags are only valid on ARM processors. 25 static const int kCpuHasARM = 0x2; 26 static const int kCpuHasNEON = 0x4; 27 // 0x8 reserved for future ARM flag. 28 29 // These flags are only valid on x86 processors. 30 static const int kCpuHasX86 = 0x10; 31 static const int kCpuHasSSE2 = 0x20; 32 static const int kCpuHasSSSE3 = 0x40; 33 static const int kCpuHasSSE41 = 0x80; 34 static const int kCpuHasSSE42 = 0x100; 35 static const int kCpuHasAVX = 0x200; 36 static const int kCpuHasAVX2 = 0x400; 37 static const int kCpuHasERMS = 0x800; 38 static const int kCpuHasFMA3 = 0x1000; 39 static const int kCpuHasAVX3 = 0x2000; 40 // 0x2000, 0x4000, 0x8000 reserved for future X86 flags. 41 42 // These flags are only valid on MIPS processors. 43 static const int kCpuHasMIPS = 0x10000; 44 static const int kCpuHasDSPR2 = 0x20000; 45 46 // Internal function used to auto-init. 47 LIBYUV_API 48 int InitCpuFlags(void); 49 50 // Internal function for parsing /proc/cpuinfo. 51 LIBYUV_API 52 int ArmCpuCaps(const char* cpuinfo_name); 53 54 // Detect CPU has SSE2 etc. 55 // Test_flag parameter should be one of kCpuHas constants above. 56 // returns non-zero if instruction set is detected 57 static __inline int TestCpuFlag(int test_flag) { 58 LIBYUV_API extern int cpu_info_; 59 return (!cpu_info_ ? InitCpuFlags() : cpu_info_) & test_flag; 60 } 61 62 // For testing, allow CPU flags to be disabled. 63 // ie MaskCpuFlags(~kCpuHasSSSE3) to disable SSSE3. 64 // MaskCpuFlags(-1) to enable all cpu specific optimizations. 65 // MaskCpuFlags(1) to disable all cpu specific optimizations. 66 LIBYUV_API 67 void MaskCpuFlags(int enable_flags); 68 69 // Low level cpuid for X86. Returns zeros on other CPUs. 70 // eax is the info type that you want. 71 // ecx is typically the cpu number, and should normally be zero. 72 LIBYUV_API 73 void CpuId(uint32 eax, uint32 ecx, uint32* cpu_info); 74 75 #ifdef __cplusplus 76 } // extern "C" 77 } // namespace libyuv 78 #endif 79 80 #endif // INCLUDE_LIBYUV_CPU_ID_H_ NOLINT 81