b113b1a382
If specified, a comma-delimited list of TCP interfaces. Interfaces will be assigned, one to each MPI process, in a round-robin fashion on each server. For example, if the list is "eth0,eth1" and four MPI processes are run on a single server, then local ranks 0 and 2 will use eth0 and local ranks 1 and 3 will use eth1. This feature is only useful for environments with virtual ethernet interfaces on the same network. For example, if eth0 and eth1 are virtual interfaces to the same NIC on the same subnet, and if the NIC provides different hardware resources to eth0 and eth1 (not just different kernel resources), some HOL blocking and congestion issues can be eased in a modest fashion. This commit was SVN r24181.
342 строки
10 KiB
C
342 строки
10 KiB
C
/*
|
|
* Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2009 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) 2010 Cisco Systems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
/**
|
|
* @file
|
|
*/
|
|
#ifndef MCA_BTL_TCP_H
|
|
#define MCA_BTL_TCP_H
|
|
|
|
#include "ompi_config.h"
|
|
#ifdef HAVE_SYS_TYPES_H
|
|
#include <sys/types.h>
|
|
#endif
|
|
#ifdef HAVE_SYS_SOCKET_H
|
|
#include <sys/socket.h>
|
|
#endif
|
|
#ifdef HAVE_NETINET_IN_H
|
|
#include <netinet/in.h>
|
|
#endif
|
|
|
|
/* Open MPI includes */
|
|
#include "opal/mca/event/event.h"
|
|
#include "ompi/class/ompi_free_list.h"
|
|
#include "ompi/mca/btl/btl.h"
|
|
#include "ompi/mca/btl/base/base.h"
|
|
#include "ompi/mca/mpool/mpool.h"
|
|
#include "ompi/mca/btl/btl.h"
|
|
#include "opal/class/opal_hash_table.h"
|
|
|
|
#define MCA_BTL_TCP_STATISTICS 0
|
|
BEGIN_C_DECLS
|
|
|
|
|
|
/**
|
|
* TCP BTL component.
|
|
*/
|
|
|
|
struct mca_btl_tcp_component_t {
|
|
mca_btl_base_component_2_0_0_t super; /**< base BTL component */
|
|
uint32_t tcp_addr_count; /**< total number of addresses */
|
|
uint32_t tcp_num_btls; /**< number of hcas available to the TCP component */
|
|
uint32_t tcp_num_links; /**< number of logical links per physical device */
|
|
struct mca_btl_tcp_module_t **tcp_btls; /**< array of available BTL modules */
|
|
struct mca_btl_tcp_proc_t* tcp_local; /**< local proc struct */
|
|
int tcp_free_list_num; /**< initial size of free lists */
|
|
int tcp_free_list_max; /**< maximum size of free lists */
|
|
int tcp_free_list_inc; /**< number of elements to alloc when growing free lists */
|
|
int tcp_endpoint_cache; /**< amount of cache on each endpoint */
|
|
opal_hash_table_t tcp_procs; /**< hash table of tcp proc structures */
|
|
opal_list_t tcp_events; /**< list of pending tcp events */
|
|
opal_mutex_t tcp_lock; /**< lock for accessing module state */
|
|
|
|
opal_event_t tcp_recv_event; /**< recv event for IPv4 listen socket */
|
|
int tcp_listen_sd; /**< IPv4 listen socket for incoming connection requests */
|
|
unsigned short tcp_listen_port; /**< IPv4 listen port */
|
|
int32_t tcp_port_min; /**< IPv4 minimum port */
|
|
int32_t tcp_port_range; /**< IPv4 port range */
|
|
#if OPAL_WANT_IPV6
|
|
opal_event_t tcp6_recv_event; /**< recv event for IPv6 listen socket */
|
|
int tcp6_listen_sd; /**< IPv6 listen socket for incoming connection requests */
|
|
unsigned short tcp6_listen_port; /**< IPv6 listen port */
|
|
int32_t tcp6_port_min; /**< IPv4 minimum port */
|
|
int32_t tcp6_port_range; /**< IPv4 port range */
|
|
#endif
|
|
/* Port range restriction */
|
|
|
|
char* tcp_if_include; /**< comma seperated list of interface to include */
|
|
char* tcp_if_exclude; /**< comma seperated list of interface to exclude */
|
|
int tcp_sndbuf; /**< socket sndbuf size */
|
|
int tcp_rcvbuf; /**< socket rcvbuf size */
|
|
int tcp_disable_family; /**< disabled AF_family */
|
|
|
|
/* free list of fragment descriptors */
|
|
ompi_free_list_t tcp_frag_eager;
|
|
ompi_free_list_t tcp_frag_max;
|
|
ompi_free_list_t tcp_frag_user;
|
|
|
|
/* Do we want to use TCP_NODELAY? */
|
|
int tcp_use_nodelay;
|
|
|
|
/* If btl_tcp_if_seq was specified, this is the one interface
|
|
(name) that we're supposed to use. */
|
|
char *tcp_if_seq;
|
|
};
|
|
typedef struct mca_btl_tcp_component_t mca_btl_tcp_component_t;
|
|
|
|
OMPI_MODULE_DECLSPEC extern mca_btl_tcp_component_t mca_btl_tcp_component;
|
|
|
|
/**
|
|
* BTL Module Interface
|
|
*/
|
|
struct mca_btl_tcp_module_t {
|
|
mca_btl_base_module_t super; /**< base BTL interface */
|
|
uint16_t tcp_ifkindex; /** <BTL kernel interface index */
|
|
#if 0
|
|
int tcp_ifindex; /**< PTL interface index */
|
|
#endif
|
|
struct sockaddr_storage tcp_ifaddr; /**< PTL interface address */
|
|
uint32_t tcp_ifmask; /**< PTL interface netmask */
|
|
opal_list_t tcp_endpoints;
|
|
#if MCA_BTL_TCP_STATISTICS
|
|
size_t tcp_bytes_sent;
|
|
size_t tcp_bytes_recv;
|
|
size_t tcp_send_handler;
|
|
#endif
|
|
};
|
|
typedef struct mca_btl_tcp_module_t mca_btl_tcp_module_t;
|
|
extern mca_btl_tcp_module_t mca_btl_tcp_module;
|
|
|
|
#if defined(__WINDOWS__)
|
|
#define CLOSE_THE_SOCKET(socket) closesocket(socket)
|
|
#else
|
|
#define CLOSE_THE_SOCKET(socket) close(socket)
|
|
#endif /* defined(__WINDOWS__) */
|
|
|
|
/**
|
|
* Register TCP component parameters with the MCA framework
|
|
*/
|
|
extern int mca_btl_tcp_component_open(void);
|
|
|
|
/**
|
|
* Any final cleanup before being unloaded.
|
|
*/
|
|
extern int mca_btl_tcp_component_close(void);
|
|
|
|
/**
|
|
* TCP component initialization.
|
|
*
|
|
* @param num_btl_modules (OUT) Number of BTLs returned in BTL array.
|
|
* @param allow_multi_user_threads (OUT) Flag indicating wether BTL supports user threads (TRUE)
|
|
* @param have_hidden_threads (OUT) Flag indicating wether BTL uses threads (TRUE)
|
|
*/
|
|
extern mca_btl_base_module_t** mca_btl_tcp_component_init(
|
|
int *num_btl_modules,
|
|
bool allow_multi_user_threads,
|
|
bool have_hidden_threads
|
|
);
|
|
|
|
/**
|
|
* TCP component control.
|
|
*/
|
|
int mca_btl_tcp_component_control(
|
|
int param,
|
|
void* value,
|
|
size_t size
|
|
);
|
|
|
|
|
|
/**
|
|
* TCP component progress.
|
|
*/
|
|
extern int mca_btl_tcp_component_progress(void);
|
|
|
|
|
|
|
|
/**
|
|
* Cleanup any resources held by the BTL.
|
|
*
|
|
* @param btl BTL instance.
|
|
* @return OMPI_SUCCESS or error status on failure.
|
|
*/
|
|
|
|
extern int mca_btl_tcp_finalize(
|
|
struct mca_btl_base_module_t* btl
|
|
);
|
|
|
|
|
|
/**
|
|
* PML->BTL notification of change in the process list.
|
|
*
|
|
* @param btl (IN)
|
|
* @param nprocs (IN) Number of processes
|
|
* @param procs (IN) Set of processes
|
|
* @param peers (OUT) Set of (optional) peer addressing info.
|
|
* @param peers (IN/OUT) Set of processes that are reachable via this BTL.
|
|
* @return OMPI_SUCCESS or error status on failure.
|
|
*
|
|
*/
|
|
|
|
extern int mca_btl_tcp_add_procs(
|
|
struct mca_btl_base_module_t* btl,
|
|
size_t nprocs,
|
|
struct ompi_proc_t **procs,
|
|
struct mca_btl_base_endpoint_t** peers,
|
|
opal_bitmap_t* reachable
|
|
);
|
|
|
|
/**
|
|
* PML->BTL notification of change in the process list.
|
|
*
|
|
* @param btl (IN) BTL instance
|
|
* @param nproc (IN) Number of processes.
|
|
* @param procs (IN) Set of processes.
|
|
* @param peers (IN) Set of peer data structures.
|
|
* @return Status indicating if cleanup was successful
|
|
*
|
|
*/
|
|
|
|
extern int mca_btl_tcp_del_procs(
|
|
struct mca_btl_base_module_t* btl,
|
|
size_t nprocs,
|
|
struct ompi_proc_t **procs,
|
|
struct mca_btl_base_endpoint_t** peers
|
|
);
|
|
|
|
|
|
/**
|
|
* Initiate an asynchronous send.
|
|
*
|
|
* @param btl (IN) BTL module
|
|
* @param endpoint (IN) BTL addressing information
|
|
* @param descriptor (IN) Description of the data to be transfered
|
|
* @param tag (IN) The tag value used to notify the peer.
|
|
*/
|
|
|
|
extern int mca_btl_tcp_send(
|
|
struct mca_btl_base_module_t* btl,
|
|
struct mca_btl_base_endpoint_t* btl_peer,
|
|
struct mca_btl_base_descriptor_t* descriptor,
|
|
mca_btl_base_tag_t tag
|
|
);
|
|
|
|
|
|
/**
|
|
* Initiate an asynchronous put.
|
|
*
|
|
* @param btl (IN) BTL module
|
|
* @param endpoint (IN) BTL addressing information
|
|
* @param descriptor (IN) Description of the data to be transferred
|
|
*/
|
|
|
|
extern int mca_btl_tcp_put(
|
|
struct mca_btl_base_module_t* btl,
|
|
struct mca_btl_base_endpoint_t* btl_peer,
|
|
struct mca_btl_base_descriptor_t* decriptor
|
|
);
|
|
|
|
|
|
/**
|
|
* Initiate an asynchronous get.
|
|
*
|
|
* @param btl (IN) BTL module
|
|
* @param endpoint (IN) BTL addressing information
|
|
* @param descriptor (IN) Description of the data to be transferred
|
|
*/
|
|
|
|
extern int mca_btl_tcp_get(
|
|
struct mca_btl_base_module_t* btl,
|
|
struct mca_btl_base_endpoint_t* btl_peer,
|
|
struct mca_btl_base_descriptor_t* decriptor
|
|
);
|
|
|
|
/**
|
|
* Allocate a descriptor with a segment of the requested size.
|
|
* Note that the BTL layer may choose to return a smaller size
|
|
* if it cannot support the request.
|
|
*
|
|
* @param btl (IN) BTL module
|
|
* @param size (IN) Request segment size.
|
|
*/
|
|
|
|
extern mca_btl_base_descriptor_t* mca_btl_tcp_alloc(
|
|
struct mca_btl_base_module_t* btl,
|
|
struct mca_btl_base_endpoint_t* endpoint,
|
|
uint8_t order,
|
|
size_t size,
|
|
uint32_t flags);
|
|
|
|
|
|
/**
|
|
* Return a segment allocated by this BTL.
|
|
*
|
|
* @param btl (IN) BTL module
|
|
* @param descriptor (IN) Allocated descriptor.
|
|
*/
|
|
|
|
extern int mca_btl_tcp_free(
|
|
struct mca_btl_base_module_t* btl,
|
|
mca_btl_base_descriptor_t* des);
|
|
|
|
|
|
/**
|
|
* Prepare a descriptor for send/rdma using the supplied
|
|
* convertor. If the convertor references data that is contigous,
|
|
* the descriptor may simply point to the user buffer. Otherwise,
|
|
* this routine is responsible for allocating buffer space and
|
|
* packing if required.
|
|
*
|
|
* @param btl (IN) BTL module
|
|
* @param endpoint (IN) BTL peer addressing
|
|
* @param convertor (IN) Data type convertor
|
|
* @param reserve (IN) Additional bytes requested by upper layer to precede user data
|
|
* @param size (IN/OUT) Number of bytes to prepare (IN), number of bytes actually prepared (OUT)
|
|
*/
|
|
|
|
mca_btl_base_descriptor_t* mca_btl_tcp_prepare_src(
|
|
struct mca_btl_base_module_t* btl,
|
|
struct mca_btl_base_endpoint_t* peer,
|
|
struct mca_mpool_base_registration_t*,
|
|
struct opal_convertor_t* convertor,
|
|
uint8_t order,
|
|
size_t reserve,
|
|
size_t* size,
|
|
uint32_t flags
|
|
);
|
|
|
|
extern mca_btl_base_descriptor_t* mca_btl_tcp_prepare_dst(
|
|
struct mca_btl_base_module_t* btl,
|
|
struct mca_btl_base_endpoint_t* peer,
|
|
struct mca_mpool_base_registration_t*,
|
|
struct opal_convertor_t* convertor,
|
|
uint8_t order,
|
|
size_t reserve,
|
|
size_t* size,
|
|
uint32_t flags);
|
|
|
|
|
|
/**
|
|
* Fault Tolerance Event Notification Function
|
|
* @param state Checkpoint Stae
|
|
* @return OMPI_SUCCESS or failure status
|
|
*/
|
|
int mca_btl_tcp_ft_event(int state);
|
|
|
|
END_C_DECLS
|
|
#endif
|