105 строки
2.4 KiB
C
105 строки
2.4 KiB
C
/*
|
|
* Copyright (c) 2007-2008 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2009 Mellanox Technologies. All rights reserved.
|
|
*
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#ifndef BTL_OPENIB_CONNECT_BASE_H
|
|
#define BTL_OPENIB_CONNECT_BASE_H
|
|
|
|
#include "connect/connect.h"
|
|
|
|
#ifdef OMPI_HAVE_RDMAOE
|
|
#define BTL_OPENIB_CONNECT_BASE_CHECK_IF_NOT_IB(btl) \
|
|
(((IBV_TRANSPORT_IB != ((btl)->device->ib_dev->transport_type)) || \
|
|
(IBV_LINK_LAYER_ETHERNET == ((btl)->ib_port_attr.link_layer))) ? \
|
|
true : false)
|
|
#else
|
|
#define BTL_OPENIB_CONNECT_BASE_CHECK_IF_NOT_IB(btl) \
|
|
((IBV_TRANSPORT_IB != ((btl)->device->ib_dev->transport_type)) ? \
|
|
true : false)
|
|
#endif
|
|
|
|
BEGIN_C_DECLS
|
|
|
|
/*
|
|
* Forward declaration to resolve circular dependency
|
|
*/
|
|
struct mca_btl_base_endpoint_t;
|
|
|
|
/*
|
|
* Open function
|
|
*/
|
|
int ompi_btl_openib_connect_base_register(void);
|
|
|
|
/*
|
|
* Component-wide CPC init
|
|
*/
|
|
int ompi_btl_openib_connect_base_init(void);
|
|
|
|
/*
|
|
* Query CPCs to see if they want to run on a specific module
|
|
*/
|
|
int ompi_btl_openib_connect_base_select_for_local_port
|
|
(mca_btl_openib_module_t *btl);
|
|
|
|
/*
|
|
* Forward reference to avoid an include file loop
|
|
*/
|
|
struct mca_btl_openib_proc_modex_t;
|
|
|
|
/*
|
|
* Select function
|
|
*/
|
|
int ompi_btl_openib_connect_base_find_match
|
|
(mca_btl_openib_module_t *btl,
|
|
struct mca_btl_openib_proc_modex_t *peer_port,
|
|
ompi_btl_openib_connect_base_module_t **local_cpc,
|
|
ompi_btl_openib_connect_base_module_data_t **remote_cpc_data);
|
|
|
|
/*
|
|
* Find a CPC's index so that we can send it in the modex
|
|
*/
|
|
int ompi_btl_openib_connect_base_get_cpc_index
|
|
(ompi_btl_openib_connect_base_component_t *cpc);
|
|
|
|
/*
|
|
* Lookup a CPC by its index (received from the modex)
|
|
*/
|
|
ompi_btl_openib_connect_base_component_t *
|
|
ompi_btl_openib_connect_base_get_cpc_byindex(uint8_t index);
|
|
|
|
/*
|
|
* Allocate a CTS frag
|
|
*/
|
|
int ompi_btl_openib_connect_base_alloc_cts(
|
|
struct mca_btl_base_endpoint_t *endpoint);
|
|
|
|
/*
|
|
* Free a CTS frag
|
|
*/
|
|
int ompi_btl_openib_connect_base_free_cts(
|
|
struct mca_btl_base_endpoint_t *endpoint);
|
|
|
|
/*
|
|
* Start a new connection to an endpoint
|
|
*/
|
|
int ompi_btl_openib_connect_base_start(
|
|
ompi_btl_openib_connect_base_module_t *cpc,
|
|
struct mca_btl_base_endpoint_t *endpoint);
|
|
|
|
|
|
/*
|
|
* Component-wide CPC finalize
|
|
*/
|
|
void ompi_btl_openib_connect_base_finalize(void);
|
|
|
|
END_C_DECLS
|
|
|
|
#endif
|