08e267b811
This commit add protection to the group, ob1, and bml endpoint lookup code. For ob1 and the bml a lock has been added. For performance reasons the lock is only held if a bml or ob1 endpoint does not exist. ompi_group_dense_lookup no uses opal_atomic_cmpset to ensure the proc is only retained by the thread that actually updates the group. Signed-off-by: Nathan Hjelm <hjelmn@lanl.gov>
103 строки
2.8 KiB
C
103 строки
2.8 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2006 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 "ompi_config.h"
|
|
#include <string.h>
|
|
|
|
#include "pml_ob1.h"
|
|
#include "pml_ob1_comm.h"
|
|
|
|
|
|
|
|
static void mca_pml_ob1_comm_proc_construct(mca_pml_ob1_comm_proc_t* proc)
|
|
{
|
|
proc->expected_sequence = 1;
|
|
proc->ompi_proc = NULL;
|
|
proc->send_sequence = 0;
|
|
OBJ_CONSTRUCT(&proc->frags_cant_match, opal_list_t);
|
|
OBJ_CONSTRUCT(&proc->specific_receives, opal_list_t);
|
|
OBJ_CONSTRUCT(&proc->unexpected_frags, opal_list_t);
|
|
}
|
|
|
|
|
|
static void mca_pml_ob1_comm_proc_destruct(mca_pml_ob1_comm_proc_t* proc)
|
|
{
|
|
OBJ_DESTRUCT(&proc->frags_cant_match);
|
|
OBJ_DESTRUCT(&proc->specific_receives);
|
|
OBJ_DESTRUCT(&proc->unexpected_frags);
|
|
if (proc->ompi_proc) {
|
|
OBJ_RELEASE(proc->ompi_proc);
|
|
}
|
|
}
|
|
|
|
|
|
OBJ_CLASS_INSTANCE(mca_pml_ob1_comm_proc_t, opal_object_t,
|
|
mca_pml_ob1_comm_proc_construct,
|
|
mca_pml_ob1_comm_proc_destruct);
|
|
|
|
|
|
static void mca_pml_ob1_comm_construct(mca_pml_ob1_comm_t* comm)
|
|
{
|
|
OBJ_CONSTRUCT(&comm->wild_receives, opal_list_t);
|
|
OBJ_CONSTRUCT(&comm->matching_lock, opal_mutex_t);
|
|
OBJ_CONSTRUCT(&comm->proc_lock, opal_mutex_t);
|
|
comm->recv_sequence = 0;
|
|
comm->procs = NULL;
|
|
comm->last_probed = 0;
|
|
comm->num_procs = 0;
|
|
}
|
|
|
|
|
|
static void mca_pml_ob1_comm_destruct(mca_pml_ob1_comm_t* comm)
|
|
{
|
|
if (NULL != comm->procs) {
|
|
for (size_t i = 0; i < comm->num_procs; ++i) {
|
|
if (comm->procs[i]) {
|
|
OBJ_RELEASE(comm->procs[i]);
|
|
}
|
|
}
|
|
|
|
free(comm->procs);
|
|
}
|
|
|
|
OBJ_DESTRUCT(&comm->wild_receives);
|
|
OBJ_DESTRUCT(&comm->matching_lock);
|
|
OBJ_DESTRUCT(&comm->proc_lock);
|
|
}
|
|
|
|
|
|
OBJ_CLASS_INSTANCE(
|
|
mca_pml_ob1_comm_t,
|
|
opal_object_t,
|
|
mca_pml_ob1_comm_construct,
|
|
mca_pml_ob1_comm_destruct);
|
|
|
|
|
|
int mca_pml_ob1_comm_init_size (mca_pml_ob1_comm_t* comm, size_t size)
|
|
{
|
|
/* send message sequence-number support - sender side */
|
|
comm->procs = (mca_pml_ob1_comm_proc_t **) calloc(size, sizeof (mca_pml_ob1_comm_proc_t *));
|
|
if(NULL == comm->procs) {
|
|
return OMPI_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
comm->num_procs = size;
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
|