149 строки
3.9 KiB
C
149 строки
3.9 KiB
C
/*
|
|
* Copyright (c) 2004-2005 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$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
#include "orte/orte_constants.h"
|
|
#include "orte/orte_types.h"
|
|
#include "orte/mca/schema/schema.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "support.h"
|
|
|
|
#include "runtime/runtime.h"
|
|
#include "opal/threads/threads.h"
|
|
#include "util/proc_info.h"
|
|
#include "util/sys_info.h"
|
|
#include "opal/util/malloc.h"
|
|
#include "opal/util/output.h"
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
#include "orte/mca/ns/ns_types.h"
|
|
#include "orte/mca/gpr/gpr.h"
|
|
#include "dps/dps.h"
|
|
|
|
#include "orte/mca/gpr/base/base.h"
|
|
#include "orte/mca/gpr/replica/api_layer/gpr_replica_api.h"
|
|
#include "orte/mca/gpr/replica/functional_layer/gpr_replica_fn.h"
|
|
#include "orte/mca/gpr/replica/communications/gpr_replica_comm.h"
|
|
#include "orte/mca/gpr/replica/transition_layer/gpr_replica_tl.h"
|
|
|
|
/* output files needed by the test */
|
|
static FILE *test_out=NULL;
|
|
|
|
/* function for exercising the registry */
|
|
void *thread_fn(opal_object_t *object);
|
|
|
|
#if !OMPI_HAVE_THREAD_SUPPORT
|
|
|
|
/* If we don't have thread support, there's no point in running this
|
|
test */
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
printf("OMPI was compiled without thread support -- skipping this test\n");
|
|
return 77;
|
|
}
|
|
|
|
#else
|
|
|
|
/* Only have the body of this test if we have thread support */
|
|
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int i, rc, num_threads;
|
|
opal_thread_t **threads;
|
|
union {
|
|
int ivalue;
|
|
void *vvalue;
|
|
} value;
|
|
|
|
/* protect against sizeof mismatches */
|
|
if (sizeof(i) > sizeof (void*)) {
|
|
fprintf(stderr, "cannot run this test on this machine\n");
|
|
exit(77);
|
|
}
|
|
|
|
test_init("test_gpr_threads");
|
|
|
|
if (getenv("TEST_WRITE_TO_FILE") != NULL) {
|
|
test_out = fopen( "test_gpr_threads", "w+" );
|
|
} else {
|
|
test_out = stderr;
|
|
}
|
|
if( test_out == NULL ) {
|
|
test_failure("gpr_threads couldn't open test file failed");
|
|
test_finalize();
|
|
exit(1);
|
|
}
|
|
|
|
if (1 < argc) { /* number of threads to run provided on cmd line */
|
|
num_threads = strtol(argv[1], NULL, 0);
|
|
} else { /* default to 2 */
|
|
num_threads = 2;
|
|
}
|
|
|
|
/* setup the runtime environment */
|
|
if (ORTE_SUCCESS != (rc = orte_init())) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
/* setup the threads */
|
|
threads = (opal_thread_t**)malloc(num_threads * sizeof(opal_thread_t*));
|
|
if (NULL == threads) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
for (i=0; i < num_threads; i++) {
|
|
threads[i] = OBJ_NEW(opal_thread_t);
|
|
if (NULL == threads[i]) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
threads[i]->t_run = thread_fn;
|
|
value.ivalue = i;
|
|
threads[i]->t_arg = value.vvalue;
|
|
}
|
|
|
|
/* run the threads */
|
|
for (i=0; i < num_threads; i++) {
|
|
if (OMPI_SUCCESS != (rc = opal_thread_start(threads[i]))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
exit(rc);
|
|
}
|
|
}
|
|
|
|
/* finalize things */
|
|
orte_finalize();
|
|
|
|
fclose( test_out );
|
|
|
|
test_finalize();
|
|
|
|
return(0);
|
|
}
|
|
|
|
void *thread_fn(opal_object_t *obj)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
#endif /* OMPI_HAVE_THREAD_SUPPORT */
|