4b6aecf82d
build / run. Only things that actually build / run right now are the asm and class tests. The mca tests probably will with a static build but that hasn't been verified This commit was SVN r4979.
110 строки
2.6 KiB
C
110 строки
2.6 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University.
|
|
* All rights reserved.
|
|
* Copyright (c) 2004-2005 The Trustees of the University of Tennessee.
|
|
* All rights reserved.
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#undef OMPI_BUILDING
|
|
#include "ompi_config.h"
|
|
|
|
#include <assert.h>
|
|
#ifdef HAVE_PTHREAD_H
|
|
#include <pthread.h>
|
|
#endif
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
#include "include/sys/atomic.h"
|
|
#include "atomic_test.h"
|
|
|
|
int atomic_verbose = 0;
|
|
|
|
struct start_info {
|
|
int tid;
|
|
int count;
|
|
ompi_lock_t *lock;
|
|
};
|
|
|
|
|
|
#if OMPI_HAVE_POSIX_THREADS
|
|
static void* atomic_spinlock_start(void* arg)
|
|
{
|
|
struct start_info *data = (struct start_info*) arg;
|
|
|
|
return (void*) (unsigned long) atomic_spinlock_test(data->lock, data->count,
|
|
data->tid);
|
|
}
|
|
#endif
|
|
|
|
|
|
int
|
|
atomic_spinlock_test_th(ompi_lock_t *lock, int count, int id, int thr_count)
|
|
{
|
|
#if OMPI_HAVE_POSIX_THREADS
|
|
pthread_t *th;
|
|
int tid, ret = 0;
|
|
struct start_info *data;
|
|
|
|
th = (pthread_t *) malloc(thr_count * sizeof(pthread_t));
|
|
if (!th) { perror("malloc"); exit(EXIT_FAILURE); }
|
|
data = (struct start_info *) malloc(thr_count * sizeof(struct start_info));
|
|
if (!th) { perror("malloc"); exit(EXIT_FAILURE); }
|
|
|
|
for (tid = 0; tid < thr_count; tid++) {
|
|
data[tid].tid = tid;
|
|
data[tid].count = count;
|
|
data[tid].lock = lock;
|
|
|
|
if (pthread_create(&th[tid], NULL, atomic_spinlock_start, (void *) &(data[tid])) != 0) {
|
|
perror("pthread_create");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
/* -- wait for the thread set to finish -- */
|
|
for (tid = 0; tid < thr_count; tid++) {
|
|
void *thread_return;
|
|
|
|
if (pthread_join(th[tid], &thread_return) != 0) {
|
|
perror("pthread_join");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
ret += (int) (unsigned long) thread_return;
|
|
}
|
|
free(data);
|
|
free(th);
|
|
|
|
return ret;
|
|
#else
|
|
return 77;
|
|
#endif
|
|
}
|
|
|
|
|
|
int
|
|
atomic_spinlock_test(ompi_lock_t *lock, int count, int id)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0 ; i < count ; ++i) {
|
|
ompi_atomic_lock(lock);
|
|
if (atomic_verbose) { printf("id %03d has the lock (lock)\n", id); }
|
|
ompi_atomic_unlock(lock);
|
|
|
|
while (ompi_atomic_trylock(lock)) { ; }
|
|
if (atomic_verbose) { printf("id %03d has the lock (trylock)\n", id); }
|
|
ompi_atomic_unlock(lock);
|
|
}
|
|
|
|
return 0;
|
|
}
|