73a943492c
If we need to use a convertor, go back to stashing that convertor in the frag and populating segments "on the fly" (in ompi_btl_usnic_module_progress_sends). Previously we would pack into a chain of chunk segments at prepare_src time, unnecessarily consuming additional memory. Reviewed-by: Jeff Squyres <jsquyres@cisco.com> Reviewed-by: Reese Faucette <rfaucett@cisco.com> This commit was SVN r29592.
100 строки
2.2 KiB
C
100 строки
2.2 KiB
C
/*
|
|
* Copyright (c) 2013 Cisco Systems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#ifndef BTL_USNIC_UTIL_H
|
|
#define BTL_USNIC_UTIL_H
|
|
|
|
#include "opal/datatype/opal_convertor.h"
|
|
|
|
#include "btl_usnic.h"
|
|
#include "btl_usnic_module.h"
|
|
|
|
/* Linux kernel fls() */
|
|
static __always_inline int fls(int x)
|
|
{
|
|
int r = 32;
|
|
|
|
if (!x)
|
|
return 0;
|
|
if (!(x & 0xffff0000u)) {
|
|
x <<= 16;
|
|
r -= 16;
|
|
}
|
|
if (!(x & 0xff000000u)) {
|
|
x <<= 8;
|
|
r -= 8;
|
|
}
|
|
if (!(x & 0xf0000000u)) {
|
|
x <<= 4;
|
|
r -= 4;
|
|
}
|
|
if (!(x & 0xc0000000u)) {
|
|
x <<= 2;
|
|
r -= 2;
|
|
}
|
|
if (!(x & 0x80000000u)) {
|
|
r -= 1;
|
|
}
|
|
return r;
|
|
}
|
|
|
|
/* a helper function that just declutters convertor packing */
|
|
static inline
|
|
int
|
|
usnic_convertor_pack_simple(
|
|
opal_convertor_t *convertor,
|
|
void *dest,
|
|
size_t max_bytes_to_pack,
|
|
size_t *bytes_packed)
|
|
{
|
|
int rc;
|
|
struct iovec iov;
|
|
uint32_t iov_count;
|
|
|
|
iov.iov_base = (IOVBASE_TYPE*)dest;
|
|
iov.iov_len = max_bytes_to_pack;
|
|
iov_count = 1;
|
|
*bytes_packed = max_bytes_to_pack;
|
|
rc = opal_convertor_pack(convertor, &iov, &iov_count, bytes_packed);
|
|
if (OPAL_UNLIKELY(rc < 0)) {
|
|
BTL_ERROR(("opal_convertor_pack error"));
|
|
abort(); /* XXX */
|
|
}
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
/*
|
|
* Safely (but abnornmally) exit this process without abort()'ing (and
|
|
* leaving a corefile).
|
|
*/
|
|
void ompi_btl_usnic_exit(void);
|
|
|
|
void ompi_btl_usnic_sprintf_mac(char *out, const uint8_t mac[6]);
|
|
|
|
void ompi_btl_usnic_sprintf_gid_mac(char *out, union ibv_gid *gid);
|
|
|
|
void ompi_btl_usnic_snprintf_bool_array(char *s, size_t slen, bool a[], size_t alen);
|
|
|
|
int ompi_btl_usnic_find_ip(ompi_btl_usnic_module_t *module, uint8_t mac[6]);
|
|
|
|
void ompi_btl_usnic_gid_to_mac(union ibv_gid *gid, uint8_t mac[6]);
|
|
|
|
void ompi_btl_usnic_dump_hex(uint8_t *addr, int len);
|
|
|
|
uint32_t ompi_btl_usnic_get_ipv4_subnet(uint32_t addrn, uint32_t cidr_len);
|
|
|
|
void ompi_btl_usnic_util_abort(const char *msg, const char *file, int line,
|
|
int ret);
|
|
|
|
size_t ompi_btl_usnic_convertor_pack_peek(const opal_convertor_t *conv,
|
|
size_t max_len);
|
|
|
|
#endif /* BTL_USNIC_UTIL_H */
|