| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | /* | ||
| 2 | * This file is part of FFmpeg. | ||
| 3 | * | ||
| 4 | * FFmpeg is free software; you can redistribute it and/or | ||
| 5 | * modify it under the terms of the GNU Lesser General Public | ||
| 6 | * License as published by the Free Software Foundation; either | ||
| 7 | * version 2.1 of the License, or (at your option) any later version. | ||
| 8 | * | ||
| 9 | * FFmpeg is distributed in the hope that it will be useful, | ||
| 10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
| 12 | * Lesser General Public License for more details. | ||
| 13 | * | ||
| 14 | * You should have received a copy of the GNU Lesser General Public | ||
| 15 | * License along with FFmpeg; if not, write to the Free Software | ||
| 16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
| 17 | */ | ||
| 18 | |||
| 19 | #include <stdio.h> | ||
| 20 | |||
| 21 | #include "config.h" | ||
| 22 | |||
| 23 | #include "libavutil/cpu.h" | ||
| 24 | #include "libavutil/avstring.h" | ||
| 25 | |||
| 26 | #if ARCH_AARCH64 | ||
| 27 | #include "libavutil/aarch64/cpu.h" | ||
| 28 | #elif ARCH_RISCV | ||
| 29 | #include "libavutil/riscv/cpu.h" | ||
| 30 | #endif | ||
| 31 | |||
| 32 | #if HAVE_UNISTD_H | ||
| 33 | #include <unistd.h> | ||
| 34 | #endif | ||
| 35 | #if !HAVE_GETOPT | ||
| 36 | #include "compat/getopt.c" | ||
| 37 | #endif | ||
| 38 | |||
| 39 | static const struct { | ||
| 40 | int flag; | ||
| 41 | const char *name; | ||
| 42 | } cpu_flag_tab[] = { | ||
| 43 | #if ARCH_AARCH64 | ||
| 44 | { AV_CPU_FLAG_ARMV8, "armv8" }, | ||
| 45 | { AV_CPU_FLAG_NEON, "neon" }, | ||
| 46 | { AV_CPU_FLAG_VFP, "vfp" }, | ||
| 47 | { AV_CPU_FLAG_DOTPROD, "dotprod" }, | ||
| 48 | { AV_CPU_FLAG_I8MM, "i8mm" }, | ||
| 49 | { AV_CPU_FLAG_SVE, "sve" }, | ||
| 50 | { AV_CPU_FLAG_SVE2, "sve2" }, | ||
| 51 | { AV_CPU_FLAG_SME, "sme" }, | ||
| 52 | { AV_CPU_FLAG_SME_I16I64, "sme_i16i64" }, | ||
| 53 | { AV_CPU_FLAG_ARM_CRC, "crc" }, | ||
| 54 | { AV_CPU_FLAG_SME2, "sme2" }, | ||
| 55 | { AV_CPU_FLAG_PMULL, "pmull" }, | ||
| 56 | { AV_CPU_FLAG_EOR3, "eor3" }, | ||
| 57 | #elif ARCH_ARM | ||
| 58 | { AV_CPU_FLAG_ARMV5TE, "armv5te" }, | ||
| 59 | { AV_CPU_FLAG_ARMV6, "armv6" }, | ||
| 60 | { AV_CPU_FLAG_ARMV6T2, "armv6t2" }, | ||
| 61 | { AV_CPU_FLAG_VFP, "vfp" }, | ||
| 62 | { AV_CPU_FLAG_VFP_VM, "vfp_vm" }, | ||
| 63 | { AV_CPU_FLAG_VFPV3, "vfpv3" }, | ||
| 64 | { AV_CPU_FLAG_NEON, "neon" }, | ||
| 65 | { AV_CPU_FLAG_SETEND, "setend" }, | ||
| 66 | #elif ARCH_PPC | ||
| 67 | { AV_CPU_FLAG_ALTIVEC, "altivec" }, | ||
| 68 | { AV_CPU_FLAG_VSX, "vsx" }, | ||
| 69 | { AV_CPU_FLAG_POWER8, "power8" }, | ||
| 70 | #elif ARCH_MIPS | ||
| 71 | { AV_CPU_FLAG_MMI, "mmi" }, | ||
| 72 | { AV_CPU_FLAG_MSA, "msa" }, | ||
| 73 | #elif ARCH_X86 | ||
| 74 | { AV_CPU_FLAG_MMX, "mmx" }, | ||
| 75 | { AV_CPU_FLAG_MMXEXT, "mmxext" }, | ||
| 76 | { AV_CPU_FLAG_SSE, "sse" }, | ||
| 77 | { AV_CPU_FLAG_SSE2, "sse2" }, | ||
| 78 | { AV_CPU_FLAG_SSE2SLOW, "sse2slow" }, | ||
| 79 | { AV_CPU_FLAG_SSE3, "sse3" }, | ||
| 80 | { AV_CPU_FLAG_SSE3SLOW, "sse3slow" }, | ||
| 81 | { AV_CPU_FLAG_SSSE3, "ssse3" }, | ||
| 82 | { AV_CPU_FLAG_ATOM, "atom" }, | ||
| 83 | { AV_CPU_FLAG_SSE4, "sse4.1" }, | ||
| 84 | { AV_CPU_FLAG_SSE42, "sse4.2" }, | ||
| 85 | { AV_CPU_FLAG_AVX, "avx" }, | ||
| 86 | { AV_CPU_FLAG_AVXSLOW, "avxslow" }, | ||
| 87 | { AV_CPU_FLAG_XOP, "xop" }, | ||
| 88 | { AV_CPU_FLAG_FMA3, "fma3" }, | ||
| 89 | { AV_CPU_FLAG_FMA4, "fma4" }, | ||
| 90 | { AV_CPU_FLAG_3DNOW, "3dnow" }, | ||
| 91 | { AV_CPU_FLAG_3DNOWEXT, "3dnowext" }, | ||
| 92 | { AV_CPU_FLAG_CMOV, "cmov" }, | ||
| 93 | { AV_CPU_FLAG_AVX2, "avx2" }, | ||
| 94 | { AV_CPU_FLAG_BMI1, "bmi1" }, | ||
| 95 | { AV_CPU_FLAG_BMI2, "bmi2" }, | ||
| 96 | { AV_CPU_FLAG_AESNI, "aesni" }, | ||
| 97 | { AV_CPU_FLAG_CLMUL, "clmul" }, | ||
| 98 | { AV_CPU_FLAG_AVX512, "avx512" }, | ||
| 99 | { AV_CPU_FLAG_AVX512ICL, "avx512icl" }, | ||
| 100 | { AV_CPU_FLAG_SLOW_GATHER, "slowgather" }, | ||
| 101 | #elif ARCH_LOONGARCH | ||
| 102 | { AV_CPU_FLAG_LSX, "lsx" }, | ||
| 103 | { AV_CPU_FLAG_LASX, "lasx" }, | ||
| 104 | #elif ARCH_RISCV | ||
| 105 | { AV_CPU_FLAG_RVI, "rvi" }, | ||
| 106 | { AV_CPU_FLAG_RVB_BASIC, "zbb" }, | ||
| 107 | { AV_CPU_FLAG_RVB, "rvb" }, | ||
| 108 | { AV_CPU_FLAG_RVV_I32, "zve32x" }, | ||
| 109 | { AV_CPU_FLAG_RVV_F32, "zve32f" }, | ||
| 110 | { AV_CPU_FLAG_RVV_I64, "zve64x" }, | ||
| 111 | { AV_CPU_FLAG_RVV_F64, "zve64d" }, | ||
| 112 | { AV_CPU_FLAG_RV_ZVBB, "zvbb" }, | ||
| 113 | { AV_CPU_FLAG_RV_MISALIGNED, "misaligned" }, | ||
| 114 | #elif ARCH_WASM | ||
| 115 | { AV_CPU_FLAG_SIMD128, "simd128" }, | ||
| 116 | #endif | ||
| 117 | { 0 } | ||
| 118 | }; | ||
| 119 | |||
| 120 | 2 | static void print_cpu_flags(int cpu_flags, const char *type) | |
| 121 | { | ||
| 122 | int i; | ||
| 123 | |||
| 124 | 2 | printf("cpu_flags(%s) = 0x%08X\n", type, cpu_flags); | |
| 125 | 2 | printf("cpu_flags_str(%s) =", type); | |
| 126 |
2/2✓ Branch 0 taken 54 times.
✓ Branch 1 taken 2 times.
|
56 | for (i = 0; cpu_flag_tab[i].flag; i++) |
| 127 |
2/2✓ Branch 0 taken 17 times.
✓ Branch 1 taken 37 times.
|
54 | if (cpu_flags & cpu_flag_tab[i].flag) |
| 128 | 17 | printf(" %s", cpu_flag_tab[i].name); | |
| 129 | 2 | printf("\n"); | |
| 130 | 2 | } | |
| 131 | |||
| 132 | |||
| 133 | 1 | int main(int argc, char **argv) | |
| 134 | { | ||
| 135 | 1 | int cpu_flags_raw = av_get_cpu_flags(); | |
| 136 | int cpu_flags_eff; | ||
| 137 | 1 | int cpu_count = av_cpu_count(); | |
| 138 | 1 | const char *threads = "auto"; | |
| 139 | int i; | ||
| 140 | |||
| 141 |
2/2✓ Branch 0 taken 27 times.
✓ Branch 1 taken 1 times.
|
28 | for(i = 0; cpu_flag_tab[i].flag; i++) { |
| 142 | 27 | unsigned tmp = 0; | |
| 143 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 27 times.
|
27 | if (av_parse_cpu_caps(&tmp, cpu_flag_tab[i].name) < 0) { |
| 144 | ✗ | fprintf(stderr, "Table missing %s\n", cpu_flag_tab[i].name); | |
| 145 | ✗ | return 4; | |
| 146 | } | ||
| 147 | } | ||
| 148 | |||
| 149 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (cpu_flags_raw < 0) |
| 150 | ✗ | return 1; | |
| 151 | |||
| 152 | 2 | for (;;) { | |
| 153 | 3 | int c = getopt(argc, argv, "c:t:"); | |
| 154 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2 times.
|
3 | if (c == -1) |
| 155 | 1 | break; | |
| 156 |
2/3✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
|
2 | switch (c) { |
| 157 | 1 | case 'c': | |
| 158 | { | ||
| 159 | 1 | unsigned flags = av_get_cpu_flags(); | |
| 160 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
|
1 | if (av_parse_cpu_caps(&flags, optarg) < 0) |
| 161 | ✗ | return 2; | |
| 162 | |||
| 163 | 1 | av_force_cpu_flags(flags); | |
| 164 | 1 | break; | |
| 165 | } | ||
| 166 | 1 | case 't': | |
| 167 | { | ||
| 168 | 1 | threads = optarg; | |
| 169 | } | ||
| 170 | } | ||
| 171 | } | ||
| 172 | |||
| 173 | 1 | cpu_flags_eff = av_get_cpu_flags(); | |
| 174 | |||
| 175 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (cpu_flags_eff < 0) |
| 176 | ✗ | return 3; | |
| 177 | |||
| 178 | 1 | print_cpu_flags(cpu_flags_raw, "raw"); | |
| 179 | 1 | print_cpu_flags(cpu_flags_eff, "effective"); | |
| 180 | 1 | printf("threads = %s (cpu_count = %d)\n", threads, cpu_count); | |
| 181 | #if ARCH_AARCH64 && HAVE_SVE | ||
| 182 | if (cpu_flags_raw & AV_CPU_FLAG_SVE) | ||
| 183 | printf("sve_vector_length = %d\n", 8 * ff_aarch64_sve_length()); | ||
| 184 | #endif | ||
| 185 | #if ARCH_AARCH64 && HAVE_SME | ||
| 186 | if (cpu_flags_raw & AV_CPU_FLAG_SME) | ||
| 187 | printf("sme_vector_length = %d\n", 8 * ff_aarch64_sme_length()); | ||
| 188 | #endif | ||
| 189 | #if ARCH_RISCV && HAVE_RVV | ||
| 190 | if (cpu_flags_raw & AV_CPU_FLAG_RVV_I32) { | ||
| 191 | size_t bytes = ff_get_rv_vlenb(); | ||
| 192 | |||
| 193 | printf("rv_vlenb = %zu (%zu bits)\n", bytes, 8 * bytes); | ||
| 194 | } | ||
| 195 | #endif | ||
| 196 | |||
| 197 | 1 | return 0; | |
| 198 | } | ||
| 199 |