9d8f634044
Now that the old sm btl has been gone for some time there was a request to rename vader to sm. This commit does just that (reluctantly). An alias has been generated so specifying vader in the btl selection variable or specifying vader parameters will continue to work. Signed-off-by: Nathan Hjelm <hjelmn@google.com>
128 строки
4.4 KiB
C
128 строки
4.4 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* 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 (c) 2006-2007 Voltaire. All rights reserved.
|
|
* Copyright (c) 2012-2016 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* Copyright (c) 2018 Triad National Security, LLC. All rights
|
|
* reserved.
|
|
* Copyright (c) 2020 Google, LLC. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
/**
|
|
* @file
|
|
*/
|
|
|
|
#ifndef MCA_BTL_SM_ENDPOINT_H
|
|
#define MCA_BTL_SM_ENDPOINT_H
|
|
|
|
#include "opal_config.h"
|
|
#include "btl_sm_xpmem.h"
|
|
|
|
#define MCA_BTL_SM_FBOX_ALIGNMENT 32
|
|
#define MCA_BTL_SM_FBOX_ALIGNMENT_MASK (MCA_BTL_SM_FBOX_ALIGNMENT - 1)
|
|
|
|
struct sm_fifo_t;
|
|
|
|
/**
|
|
* An abstraction that represents a connection to a endpoint process.
|
|
* An instance of mca_ptl_base_endpoint_t is associated w/ each process
|
|
* and BTL pair at startup.
|
|
*/
|
|
|
|
struct mca_btl_sm_fbox_t;
|
|
|
|
typedef struct mca_btl_base_endpoint_t {
|
|
opal_list_item_t super;
|
|
|
|
/* per peer buffers */
|
|
struct {
|
|
unsigned char *buffer; /**< starting address of peer's fast box out */
|
|
uint32_t *startp;
|
|
unsigned int start;
|
|
uint16_t seq;
|
|
} fbox_in;
|
|
|
|
struct {
|
|
unsigned char *buffer; /**< starting address of peer's fast box in */
|
|
uint32_t *startp; /**< pointer to location storing start offset */
|
|
unsigned int start, end;
|
|
uint16_t seq;
|
|
opal_free_list_item_t *fbox; /**< fast-box free list item */
|
|
} fbox_out;
|
|
|
|
int32_t peer_smp_rank; /**< my peer's SMP process rank. Used for accessing
|
|
* SMP specfic data structures. */
|
|
opal_atomic_size_t send_count; /**< number of fragments sent to this peer */
|
|
char *segment_base; /**< start of the peer's segment (in the address space
|
|
* of this process) */
|
|
|
|
struct sm_fifo_t *fifo; /**< */
|
|
|
|
opal_mutex_t lock; /**< lock to protect endpoint structures from concurrent
|
|
* access */
|
|
|
|
union {
|
|
#if OPAL_BTL_SM_HAVE_XPMEM
|
|
struct {
|
|
xpmem_apid_t apid; /**< xpmem apid for remote peer */
|
|
uintptr_t address_max; /**< largest address that can be attached */
|
|
} xpmem;
|
|
#endif
|
|
struct {
|
|
pid_t pid; /**< pid of remote peer (used for CMA) */
|
|
opal_shmem_ds_t *seg_ds; /**< stored segment information for detach */
|
|
} other;
|
|
} segment_data;
|
|
|
|
opal_mutex_t pending_frags_lock; /**< protect pending_frags */
|
|
opal_list_t pending_frags; /**< fragments pending fast box space */
|
|
bool waiting; /**< endpoint is on the component wait list */
|
|
} mca_btl_base_endpoint_t;
|
|
|
|
typedef mca_btl_base_endpoint_t mca_btl_sm_endpoint_t;
|
|
|
|
OBJ_CLASS_DECLARATION(mca_btl_sm_endpoint_t);
|
|
|
|
static inline void mca_btl_sm_endpoint_setup_fbox_recv (struct mca_btl_base_endpoint_t *endpoint, void *base)
|
|
{
|
|
endpoint->fbox_in.startp = (uint32_t *) base;
|
|
endpoint->fbox_in.start = MCA_BTL_SM_FBOX_ALIGNMENT;
|
|
endpoint->fbox_in.seq = 0;
|
|
opal_atomic_wmb ();
|
|
endpoint->fbox_in.buffer = base;
|
|
}
|
|
|
|
static inline void mca_btl_sm_endpoint_setup_fbox_send (struct mca_btl_base_endpoint_t *endpoint, opal_free_list_item_t *fbox)
|
|
{
|
|
void *base = fbox->ptr;
|
|
|
|
endpoint->fbox_out.start = MCA_BTL_SM_FBOX_ALIGNMENT;
|
|
endpoint->fbox_out.end = MCA_BTL_SM_FBOX_ALIGNMENT;
|
|
endpoint->fbox_out.startp = (uint32_t *) base;
|
|
endpoint->fbox_out.startp[0] = MCA_BTL_SM_FBOX_ALIGNMENT;
|
|
endpoint->fbox_out.seq = 0;
|
|
endpoint->fbox_out.fbox = fbox;
|
|
|
|
/* zero out the first header in the fast box */
|
|
memset ((char *) base + MCA_BTL_SM_FBOX_ALIGNMENT, 0, MCA_BTL_SM_FBOX_ALIGNMENT);
|
|
|
|
opal_atomic_wmb ();
|
|
endpoint->fbox_out.buffer = base;
|
|
}
|
|
|
|
#endif /* MCA_BTL_SM_ENDPOINT_H */
|