80 строки
2.7 KiB
C
80 строки
2.7 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2009-2012 Oak Ridge National Laboratory. All rights reserved.
|
||
|
|
* Copyright (c) 2009-2012 Mellanox Technologies. All rights reserved.
|
||
|
|
* $COPYRIGHT$
|
||
|
|
*
|
||
|
|
* Additional copyrights may follow
|
||
|
|
*
|
||
|
|
* $HEADER$
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef MCA_BCOL_PTPCOLL_UTILS_H
|
||
|
|
#define MCA_BCOL_PTPCOLL_UTILS_H
|
||
|
|
|
||
|
|
#include "ompi_config.h"
|
||
|
|
#include "orte/util/proc_info.h"
|
||
|
|
#include "orte/util/name_fns.h"
|
||
|
|
#include "orte/runtime/orte_globals.h"
|
||
|
|
|
||
|
|
BEGIN_C_DECLS
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Return closet power of K, for the number
|
||
|
|
*/
|
||
|
|
int pow_k_calc(int k, int number, int *out_number);
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Communicator rank to group index conversion function for K-nomial tree.
|
||
|
|
*/
|
||
|
|
int get_group_index_for_k_nomial(int my_group_index, int comm_source, int radix, int group_size, int *group_array);
|
||
|
|
|
||
|
|
/* the same like above, just more information on return */
|
||
|
|
int get_group_index_and_distance_for_k_nomial(int my_group_index, int comm_source, int radix,
|
||
|
|
int group_size, int *group_array, int *pow_distance);
|
||
|
|
|
||
|
|
int get_group_index_and_distance_for_binomial(int my_group_index, int comm_source,
|
||
|
|
int group_size, int *group_array, int *pow_distance);
|
||
|
|
/*
|
||
|
|
* Error and debug Macros/Functions
|
||
|
|
*/
|
||
|
|
static inline int mca_bcol_ptpcoll_err(const char* fmt, ...)
|
||
|
|
{
|
||
|
|
va_list list;
|
||
|
|
int ret;
|
||
|
|
|
||
|
|
va_start(list, fmt);
|
||
|
|
ret = vfprintf(stderr, fmt, list);
|
||
|
|
va_end(list);
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
#define PTPCOLL_ERROR(args) \
|
||
|
|
do { \
|
||
|
|
mca_bcol_ptpcoll_err("[%s]%s[%s:%d:%s] PTPCOLL ", \
|
||
|
|
orte_process_info.nodename, \
|
||
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME), \
|
||
|
|
__FILE__, __LINE__, __func__); \
|
||
|
|
mca_bcol_ptpcoll_err args; \
|
||
|
|
mca_bcol_ptpcoll_err("\n"); \
|
||
|
|
} while(0)
|
||
|
|
|
||
|
|
#if OPAL_ENABLE_DEBUG
|
||
|
|
#define PTPCOLL_VERBOSE(level, args) \
|
||
|
|
do { \
|
||
|
|
if (mca_bcol_ptpcoll_component.verbose >= level) { \
|
||
|
|
mca_bcol_ptpcoll_err("[%s]%s[%s:%d:%s] PTPCOLL ", \
|
||
|
|
orte_process_info.nodename, \
|
||
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME), \
|
||
|
|
__FILE__, __LINE__, __func__); \
|
||
|
|
mca_bcol_ptpcoll_err args; \
|
||
|
|
mca_bcol_ptpcoll_err("\n"); \
|
||
|
|
} \
|
||
|
|
} while(0)
|
||
|
|
#else
|
||
|
|
#define PTPCOLL_VERBOSE(level, args)
|
||
|
|
#endif
|
||
|
|
|
||
|
|
END_C_DECLS
|
||
|
|
|
||
|
|
#endif
|