diff options
author | Thilo Borgmann <thilo.borgmann@mail.de> | 2021-06-06 15:15:00 +0200 |
---|---|---|
committer | Thilo Borgmann <thilo.borgmann@mail.de> | 2021-07-16 10:06:10 +0200 |
commit | 87951dcbe775b349a671b9ac2e6ac5c38aee0e79 (patch) | |
tree | 7e0b5e7990907bf96b6db25e498f6e03307cdd1e /libavutil/cpu.c | |
parent | b7266302a40ba48fea7a5644f08623159b3dcac7 (diff) |
lavu/cpu.c: Add av_force_cpu_count() to override auto-detection.
Diffstat (limited to 'libavutil/cpu.c')
-rw-r--r-- | libavutil/cpu.c | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/libavutil/cpu.c b/libavutil/cpu.c index 8960415d00..eae1485f36 100644 --- a/libavutil/cpu.c +++ b/libavutil/cpu.c @@ -48,6 +48,7 @@ #endif static atomic_int cpu_flags = ATOMIC_VAR_INIT(-1); +static atomic_int cpu_count = ATOMIC_VAR_INIT(-1); static int get_cpu_flags(void) { @@ -186,6 +187,7 @@ int av_cpu_count(void) static atomic_int printed = ATOMIC_VAR_INIT(0); int nb_cpus = 1; + int count = 0; #if HAVE_WINRT SYSTEM_INFO sysinfo; #endif @@ -224,9 +226,20 @@ int av_cpu_count(void) if (!atomic_exchange_explicit(&printed, 1, memory_order_relaxed)) av_log(NULL, AV_LOG_DEBUG, "detected %d logical cores\n", nb_cpus); + count = atomic_load_explicit(&cpu_count, memory_order_relaxed); + + if (count > 0) { + nb_cpus = count; + av_log(NULL, AV_LOG_DEBUG, "overriding to %d logical cores\n", nb_cpus); + } + return nb_cpus; } +void av_force_cpu_count(int count){ + atomic_store_explicit(&cpu_count, count, memory_order_relaxed); +} + size_t av_cpu_max_align(void) { if (ARCH_MIPS) |