diff options
Diffstat (limited to 'sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c')
-rw-r--r-- | sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c | 280 |
1 files changed, 0 insertions, 280 deletions
diff --git a/sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c b/sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c deleted file mode 100644 index 91ed987563..0000000000 --- a/sysdeps/unix/sysv/linux/tst-skeleton-thread-affinity.c +++ /dev/null @@ -1,280 +0,0 @@ -/* Generic test for CPU affinity functions, multi-threaded variant. - Copyright (C) 2015-2017 Free Software Foundation, Inc. - This file is part of the GNU C Library. - - The GNU C Library is free software; you can redistribute it and/or - modify it under the terms of the GNU Lesser General Public - License as published by the Free Software Foundation; either - version 2.1 of the License, or (at your option) any later version. - - The GNU C Library is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - Lesser General Public License for more details. - - You should have received a copy of the GNU Lesser General Public - License along with the GNU C Library; if not, see - <http://www.gnu.org/licenses/>. */ - -/* Before including this file, a test has to declare the helper - getaffinity and setaffinity functions described in - tst-skeleton-affinity.c, which is included below. */ - -#include <errno.h> -#include <pthread.h> -#include <stdbool.h> -#include <stdlib.h> -#include <sys/time.h> - -struct conf; -static bool early_test (struct conf *); - -/* Arbitrary run time for each pass. */ -#define PASS_TIMEOUT 2 - -/* There are two passes (one with sched_yield, one without), and we - double the timeout to be on the safe side. */ -#define TIMEOUT (2 * PASS_TIMEOUT * 2) - -#include "tst-skeleton-affinity.c" - -/* 0 if still running, 1 of stopping requested. */ -static int still_running; - -/* 0 if no scheduling failures, 1 if failures are encountered. */ -static int failed; - -static void * -thread_burn_one_cpu (void *closure) -{ - int cpu = (uintptr_t) closure; - while (__atomic_load_n (&still_running, __ATOMIC_RELAXED) == 0) - { - int current = sched_getcpu (); - if (sched_getcpu () != cpu) - { - printf ("error: Pinned thread %d ran on impossible cpu %d\n", - cpu, current); - __atomic_store_n (&failed, 1, __ATOMIC_RELAXED); - /* Terminate early. */ - __atomic_store_n (&still_running, 1, __ATOMIC_RELAXED); - } - } - return NULL; -} - -struct burn_thread -{ - pthread_t self; - struct conf *conf; - cpu_set_t *initial_set; - cpu_set_t *seen_set; - int thread; -}; - -static void * -thread_burn_any_cpu (void *closure) -{ - struct burn_thread *param = closure; - - /* Schedule this thread around a bit to see if it lands on another - CPU. Run this for 2 seconds, once with sched_yield, once - without. */ - for (int pass = 1; pass <= 2; ++pass) - { - time_t start = time (NULL); - while (time (NULL) - start <= PASS_TIMEOUT) - { - int cpu = sched_getcpu (); - if (cpu > param->conf->last_cpu - || !CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (param->conf->set_size), - param->initial_set)) - { - printf ("error: Unpinned thread %d ran on impossible CPU %d\n", - param->thread, cpu); - __atomic_store_n (&failed, 1, __ATOMIC_RELAXED); - return NULL; - } - CPU_SET_S (cpu, CPU_ALLOC_SIZE (param->conf->set_size), - param->seen_set); - if (pass == 1) - sched_yield (); - } - } - return NULL; -} - -static void -stop_and_join_threads (struct conf *conf, cpu_set_t *set, - pthread_t *pinned_first, pthread_t *pinned_last, - struct burn_thread *other_first, - struct burn_thread *other_last) -{ - __atomic_store_n (&still_running, 1, __ATOMIC_RELAXED); - for (pthread_t *p = pinned_first; p < pinned_last; ++p) - { - int cpu = p - pinned_first; - if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), set)) - continue; - - int ret = pthread_join (*p, NULL); - if (ret != 0) - { - printf ("error: Failed to join thread %d: %s\n", cpu, strerror (ret)); - fflush (stdout); - /* Cannot shut down cleanly with threads still running. */ - abort (); - } - } - - for (struct burn_thread *p = other_first; p < other_last; ++p) - { - int cpu = p - other_first; - if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), set)) - continue; - - int ret = pthread_join (p->self, NULL); - if (ret != 0) - { - printf ("error: Failed to join thread %d: %s\n", cpu, strerror (ret)); - fflush (stdout); - /* Cannot shut down cleanly with threads still running. */ - abort (); - } - } -} - -/* Tries to check that the initial set of CPUs is complete and that - the main thread will not run on any other threads. */ -static bool -early_test (struct conf *conf) -{ - pthread_t *pinned_threads - = calloc (conf->last_cpu + 1, sizeof (*pinned_threads)); - struct burn_thread *other_threads - = calloc (conf->last_cpu + 1, sizeof (*other_threads)); - cpu_set_t *initial_set = CPU_ALLOC (conf->set_size); - cpu_set_t *scratch_set = CPU_ALLOC (conf->set_size); - - if (pinned_threads == NULL || other_threads == NULL - || initial_set == NULL || scratch_set == NULL) - { - puts ("error: Memory allocation failure"); - return false; - } - if (getaffinity (CPU_ALLOC_SIZE (conf->set_size), initial_set) < 0) - { - printf ("error: pthread_getaffinity_np failed: %m\n"); - return false; - } - for (int cpu = 0; cpu <= conf->last_cpu; ++cpu) - { - if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), initial_set)) - continue; - other_threads[cpu].conf = conf; - other_threads[cpu].initial_set = initial_set; - other_threads[cpu].thread = cpu; - other_threads[cpu].seen_set = CPU_ALLOC (conf->set_size); - if (other_threads[cpu].seen_set == NULL) - { - puts ("error: Memory allocation failure"); - return false; - } - CPU_ZERO_S (CPU_ALLOC_SIZE (conf->set_size), - other_threads[cpu].seen_set); - } - - pthread_attr_t attr; - int ret = pthread_attr_init (&attr); - if (ret != 0) - { - printf ("error: pthread_attr_init failed: %s\n", strerror (ret)); - return false; - } - - /* Spawn a thread pinned to each available CPU. */ - for (int cpu = 0; cpu <= conf->last_cpu; ++cpu) - { - if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), initial_set)) - continue; - CPU_ZERO_S (CPU_ALLOC_SIZE (conf->set_size), scratch_set); - CPU_SET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), scratch_set); - ret = pthread_attr_setaffinity_np - (&attr, CPU_ALLOC_SIZE (conf->set_size), scratch_set); - if (ret != 0) - { - printf ("error: pthread_attr_setaffinity_np for CPU %d failed: %s\n", - cpu, strerror (ret)); - stop_and_join_threads (conf, initial_set, - pinned_threads, pinned_threads + cpu, - NULL, NULL); - return false; - } - ret = pthread_create (pinned_threads + cpu, &attr, - thread_burn_one_cpu, (void *) (uintptr_t) cpu); - if (ret != 0) - { - printf ("error: pthread_create for CPU %d failed: %s\n", - cpu, strerror (ret)); - stop_and_join_threads (conf, initial_set, - pinned_threads, pinned_threads + cpu, - NULL, NULL); - return false; - } - } - - /* Spawn another set of threads running on all CPUs. */ - for (int cpu = 0; cpu <= conf->last_cpu; ++cpu) - { - if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), initial_set)) - continue; - ret = pthread_create (&other_threads[cpu].self, NULL, - thread_burn_any_cpu, other_threads + cpu); - if (ret != 0) - { - printf ("error: pthread_create for thread %d failed: %s\n", - cpu, strerror (ret)); - stop_and_join_threads (conf, initial_set, - pinned_threads, - pinned_threads + conf->last_cpu + 1, - other_threads, other_threads + cpu); - return false; - } - } - - /* Main thread. */ - struct burn_thread main_thread; - main_thread.conf = conf; - main_thread.initial_set = initial_set; - main_thread.seen_set = scratch_set; - main_thread.thread = -1; - CPU_ZERO_S (CPU_ALLOC_SIZE (conf->set_size), main_thread.seen_set); - thread_burn_any_cpu (&main_thread); - stop_and_join_threads (conf, initial_set, - pinned_threads, - pinned_threads + conf->last_cpu + 1, - other_threads, other_threads + conf->last_cpu + 1); - - printf ("info: Main thread ran on %d CPU(s) of %d available CPU(s)\n", - CPU_COUNT_S (CPU_ALLOC_SIZE (conf->set_size), scratch_set), - CPU_COUNT_S (CPU_ALLOC_SIZE (conf->set_size), initial_set)); - CPU_ZERO_S (CPU_ALLOC_SIZE (conf->set_size), scratch_set); - for (int cpu = 0; cpu <= conf->last_cpu; ++cpu) - { - if (!CPU_ISSET_S (cpu, CPU_ALLOC_SIZE (conf->set_size), initial_set)) - continue; - CPU_OR_S (CPU_ALLOC_SIZE (conf->set_size), - scratch_set, scratch_set, other_threads[cpu].seen_set); - CPU_FREE (other_threads[cpu].seen_set); - } - printf ("info: Other threads ran on %d CPU(s)\n", - CPU_COUNT_S (CPU_ALLOC_SIZE (conf->set_size), scratch_set));; - - - pthread_attr_destroy (&attr); - CPU_FREE (scratch_set); - CPU_FREE (initial_set); - free (pinned_threads); - free (other_threads); - return failed == 0; -} |