1
1
openmpi/opal/threads/condition.h
Josh Hursey 99144db970 Improve checkpoint/restart support by allowing a checkpoint to progress when the process is *not* in the MPI library. This involves creating a separate thread for polling for a checkpoint request. This thread is active when the MPI process is not in the MPI library, and paused when the MPI process is in the library.
Some MPI C interface files saw some spacing changes to conform to the coding standards of Open MPI.

Changed MPI C interface files to use {{{OPAL_CR_ENTER_LIBRARY()}}} and {{{OPAL_CR_EXIT_LIBRARY()}}} instead of just {{{OPAL_CR_TEST_CHECKPOINT_READY()}}}. This will allow the checkpoint/restart system more flexibility in how it is to behave.

Fixed the configure check for {{{--enable-ft-thread}}} so it has a know dependance on {{{--enable-mpi-thread}}} (and/or {{{--enable-progress-thread}}}).

Added a line for Checkpoint/Restart support to {{{ompi_info}}}.

Added some options to choose at runtime whether or not to use the checkpoint polling thread. By default, if the user asked for it to be compiled in, then it is used. But some users will want the ability to toggle its use at runtime.

There are still some places for improvement, but the feature works correctly. As always with Checkpoint/Restart, it is compiled out unless explicitly asked for at configure time. Further, if it was configured in, then it is not used unless explicitly asked for by the user at runtime.

This commit was SVN r17516.
2008-02-19 22:15:52 +00:00

216 строки
6.2 KiB
C

/*
* Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2005 The University of Tennessee and The University
* of Tennessee Research Foundation. All rights
* reserved.
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* Copyright (c) 2007 Los Alamos National Security, LLC. All rights
* reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#ifndef OPAL_CONDITION_SPINLOCK_H
#define OPAL_CONDITION_SPINLOCK_H
#include "opal_config.h"
#ifdef HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
#ifdef HAVE_TIME_H
#include <time.h>
#endif
#if OMPI_HAVE_POSIX_THREADS
#include <pthread.h>
#elif OMPI_HAVE_SOLARIS_THREADS
#include <thread.h>
#include <synch.h>
#endif
#include "opal/threads/condition.h"
#include "opal/threads/mutex.h"
#include "opal/runtime/opal_progress.h"
#include "opal/runtime/opal_cr.h"
BEGIN_C_DECLS
/*
* Combine pthread support w/ polled progress to allow run-time selection
* of threading vs. non-threading progress.
*/
struct opal_condition_t {
opal_object_t super;
volatile int c_waiting;
volatile int c_signaled;
#if OMPI_HAVE_POSIX_THREADS
pthread_cond_t c_cond;
#elif OMPI_HAVE_SOLARIS_THREADS
cond_t c_cond;
#endif
};
typedef struct opal_condition_t opal_condition_t;
OPAL_DECLSPEC OBJ_CLASS_DECLARATION(opal_condition_t);
static inline int opal_condition_wait(opal_condition_t *c, opal_mutex_t *m)
{
int rc = 0;
c->c_waiting++;
#if OMPI_ENABLE_DEBUG && !OMPI_HAVE_THREAD_SUPPORT
if (opal_mutex_check_locks && 0 == m->m_lock_debug) { \
opal_output(0, "Warning -- mutex not locked in condition_wait"); \
} \
m->m_lock_debug--;
#endif
if (opal_using_threads()) {
#if OMPI_HAVE_POSIX_THREADS && OMPI_ENABLE_PROGRESS_THREADS
rc = pthread_cond_wait(&c->c_cond, &m->m_lock_pthread);
#elif OMPI_HAVE_SOLARIS_THREADS && OMPI_ENABLE_PROGRESS_THREADS
rc = cond_wait(&c->c_cond, &m->m_lock_solaris);
#else
if (c->c_signaled) {
c->c_waiting--;
opal_mutex_unlock(m);
opal_progress();
opal_mutex_lock(m);
return 0;
}
while (c->c_signaled == 0) {
opal_mutex_unlock(m);
opal_progress();
OPAL_CR_TEST_CHECKPOINT_READY_STALL();
opal_mutex_lock(m);
}
#endif
} else {
while (c->c_signaled == 0) {
opal_progress();
OPAL_CR_TEST_CHECKPOINT_READY_STALL();
}
}
#if OMPI_ENABLE_DEBUG && !OMPI_HAVE_THREAD_SUPPORT
m->m_lock_debug++;
#endif
c->c_signaled--;
c->c_waiting--;
return rc;
}
static inline int opal_condition_timedwait(opal_condition_t *c,
opal_mutex_t *m,
const struct timespec *abstime)
{
struct timeval tv;
struct timeval absolute;
int rc = 0;
#if OMPI_ENABLE_DEBUG && !OMPI_HAVE_THREAD_SUPPORT
if (opal_mutex_check_locks && 0 == m->m_lock_debug) { \
opal_output(0, "Warning -- mutex not locked in condition_wait"); \
} \
m->m_lock_debug--;
#endif
c->c_waiting++;
if (opal_using_threads()) {
#if OMPI_HAVE_POSIX_THREADS && OMPI_ENABLE_PROGRESS_THREADS
rc = pthread_cond_timedwait(&c->c_cond, &m->m_lock_pthread, abstime);
#elif OMPI_HAVE_SOLARIS_THREADS && OMPI_ENABLE_PROGRESS_THREADS
/* deal with const-ness */
timestruc_t to;
to.tv_sec = abstime->tv_sec;
to.tv_nsec = abstime->tv_nsec;
rc = cond_timedwait(&c->c_cond, &m->m_lock_solaris, &to);
#else
absolute.tv_sec = abstime->tv_sec;
absolute.tv_usec = abstime->tv_nsec * 1000;
gettimeofday(&tv,NULL);
if (c->c_signaled == 0) {
do {
opal_mutex_unlock(m);
opal_progress();
gettimeofday(&tv,NULL);
opal_mutex_lock(m);
} while (c->c_signaled == 0 &&
(tv.tv_sec <= absolute.tv_sec ||
(tv.tv_sec == absolute.tv_sec && tv.tv_usec < absolute.tv_usec)));
}
#endif
} else {
absolute.tv_sec = abstime->tv_sec;
absolute.tv_usec = abstime->tv_nsec * 1000;
gettimeofday(&tv,NULL);
if (c->c_signaled == 0) {
do {
opal_progress();
gettimeofday(&tv,NULL);
} while (c->c_signaled == 0 &&
(tv.tv_sec <= absolute.tv_sec ||
(tv.tv_sec == absolute.tv_sec && tv.tv_usec < absolute.tv_usec)));
}
}
#if OMPI_ENABLE_DEBUG && !OMPI_HAVE_THREAD_SUPPORT
m->m_lock_debug++;
#endif
if (c->c_signaled != 0) c->c_signaled--;
c->c_waiting--;
return rc;
}
static inline int opal_condition_signal(opal_condition_t *c)
{
if (c->c_waiting) {
c->c_signaled++;
#if OMPI_HAVE_POSIX_THREADS && OMPI_ENABLE_PROGRESS_THREADS
if(opal_using_threads()) {
pthread_cond_signal(&c->c_cond);
}
#elif OMPI_HAVE_SOLARIS_THREADS && OMPI_ENABLE_PROGRESS_THREADS
if(opal_using_threads()) {
cond_signal(&c->c_cond);
}
#endif
}
return 0;
}
static inline int opal_condition_broadcast(opal_condition_t *c)
{
c->c_signaled = c->c_waiting;
#if OMPI_HAVE_POSIX_THREADS && OMPI_ENABLE_PROGRESS_THREADS
if(opal_using_threads()) {
if( 1 == c->c_waiting ) {
pthread_cond_signal(&c->c_cond);
} else {
pthread_cond_broadcast(&c->c_cond);
}
}
#elif OMPI_HAVE_SOLARIS_THREADS && OMPI_ENABLE_PROGRESS_THREADS
if(opal_using_threads()) {
cond_broadcast(&c->c_cond);
}
#endif
return 0;
}
END_C_DECLS
#endif