154 lines
4.8 KiB
C
154 lines
4.8 KiB
C
/* Copyright (C) 2015-2022 Free Software Foundation, Inc.
|
|
Contributed by Sebastian Huber <sebastian.huber@embedded-brains.de>.
|
|
|
|
This file is part of the GNU Offloading and Multi Processing Library
|
|
(libgomp).
|
|
|
|
Libgomp is free software; you can redistribute it and/or modify it
|
|
under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 3, or (at your option)
|
|
any later version.
|
|
|
|
Libgomp 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 General Public License for
|
|
more details.
|
|
|
|
Under Section 7 of GPL version 3, you are granted additional
|
|
permissions described in the GCC Runtime Library Exception, version
|
|
3.1, as published by the Free Software Foundation.
|
|
|
|
You should have received a copy of the GNU General Public License and
|
|
a copy of the GCC Runtime Library Exception along with this program;
|
|
see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
|
|
<http://www.gnu.org/licenses/>. */
|
|
|
|
/* This is the RTEMS implementation of the thread pool management
|
|
for libgomp. This type is private to the library. */
|
|
|
|
#ifndef GOMP_POOL_H
|
|
#define GOMP_POOL_H 1
|
|
|
|
#include "libgomp.h"
|
|
#include <sys/lock.h>
|
|
#include <string.h>
|
|
|
|
/* For each scheduler instance there may be a thread pool reservoir
|
|
to limit the number of thread pools used by the OpenMP master threads of this
|
|
scheduler instance. The reservoirs are configured via the
|
|
GOMP_RTEMS_THREAD_POOLS environment variable. */
|
|
struct gomp_thread_pool_reservoir {
|
|
gomp_sem_t available;
|
|
pthread_spinlock_t lock;
|
|
size_t index;
|
|
int priority;
|
|
struct gomp_thread_pool *pools[];
|
|
};
|
|
|
|
struct gomp_tls_rtems_data {
|
|
struct gomp_thread_pool_reservoir *thread_pool_reservoir;
|
|
};
|
|
|
|
extern struct gomp_thread_pool_reservoir **gomp_thread_pool_reservoirs;
|
|
|
|
extern __thread struct gomp_tls_rtems_data gomp_tls_rtems_data;
|
|
|
|
static inline struct gomp_thread_pool_reservoir *
|
|
gomp_get_thread_pool_reservoir (void)
|
|
{
|
|
struct gomp_thread_pool_reservoir *res =
|
|
gomp_tls_rtems_data.thread_pool_reservoir;
|
|
|
|
if (res == NULL && gomp_thread_pool_reservoirs != NULL)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
thr->thread_pool = gomp_malloc_cleared (sizeof (*thr->thread_pool));
|
|
res = gomp_thread_pool_reservoirs[_Sched_Index ()];
|
|
gomp_tls_rtems_data.thread_pool_reservoir = res;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
static inline struct gomp_thread_pool *
|
|
gomp_get_own_thread_pool (struct gomp_thread *thr, unsigned nthreads)
|
|
{
|
|
struct gomp_thread_pool *pool = thr->thread_pool;
|
|
if (__builtin_expect (pool == NULL, 0))
|
|
{
|
|
pool = gomp_malloc_cleared (sizeof (*pool));
|
|
pool->threads_busy = nthreads;
|
|
thr->thread_pool = pool;
|
|
}
|
|
return pool;
|
|
}
|
|
|
|
static inline struct gomp_thread_pool *
|
|
gomp_get_thread_pool (struct gomp_thread *thr, unsigned nthreads)
|
|
{
|
|
struct gomp_thread_pool *pool;
|
|
struct gomp_thread_pool_reservoir *res;
|
|
|
|
if (__builtin_expect (thr->thread_pool == NULL, 0))
|
|
pthread_setspecific (gomp_thread_destructor, thr);
|
|
|
|
res = gomp_get_thread_pool_reservoir ();
|
|
if (res != NULL)
|
|
{
|
|
gomp_sem_wait (&res->available);
|
|
pthread_spin_lock (&res->lock);
|
|
pool = res->pools[--res->index];
|
|
pthread_spin_unlock (&res->lock);
|
|
pool->threads_busy = nthreads;
|
|
thr->thread_pool = pool;
|
|
}
|
|
else
|
|
pool = gomp_get_own_thread_pool (thr, nthreads);
|
|
|
|
return pool;
|
|
}
|
|
|
|
static inline void
|
|
gomp_release_thread_pool (struct gomp_thread_pool *pool)
|
|
{
|
|
struct gomp_thread_pool_reservoir *res =
|
|
gomp_tls_rtems_data.thread_pool_reservoir;
|
|
if (res != NULL)
|
|
{
|
|
pthread_spin_lock (&res->lock);
|
|
res->pools[res->index++] = pool;
|
|
pthread_spin_unlock (&res->lock);
|
|
gomp_sem_post (&res->available);
|
|
}
|
|
}
|
|
|
|
static inline pthread_attr_t *
|
|
gomp_adjust_thread_attr (pthread_attr_t *attr, pthread_attr_t *mutable_attr)
|
|
{
|
|
struct gomp_thread_pool_reservoir *res = gomp_get_thread_pool_reservoir ();
|
|
if (res != NULL && res->priority > 0)
|
|
{
|
|
struct sched_param param;
|
|
int err;
|
|
if (attr != mutable_attr)
|
|
{
|
|
attr = mutable_attr;
|
|
pthread_attr_init (attr);
|
|
}
|
|
memset (¶m, 0, sizeof (param));
|
|
param.sched_priority = res->priority;
|
|
err = pthread_attr_setschedparam (attr, ¶m);
|
|
if (err != 0)
|
|
gomp_fatal ("Thread attribute set scheduler parameters failed: %s", strerror (err));
|
|
err = pthread_attr_setschedpolicy (attr, SCHED_FIFO);
|
|
if (err != 0)
|
|
gomp_fatal ("Thread attribute set scheduler policy failed: %s", strerror (err));
|
|
err = pthread_attr_setinheritsched (attr, PTHREAD_EXPLICIT_SCHED);
|
|
if (err != 0)
|
|
gomp_fatal ("Thread attribute set explicit scheduler failed: %s", strerror (err));
|
|
}
|
|
return attr;
|
|
}
|
|
|
|
#endif /* GOMP_POOL_H */
|