#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include "ucx_info.h"
#include <ucs/config/parser.h>
#include <ucs/config/global_opts.h>
#include <ucm/api/ucm.h>
#include <getopt.h>
#include <stdlib.h>
#include <string.h>
#include <limits.h>
static void usage() {
printf("Usage: ucx_info [options]\n");
printf("At least one of the following options has to be set:\n");
printf(" -v Show version information\n");
printf(" -d Show devices and transports\n");
printf(" -b Show build configuration\n");
printf(" -y Show type and structures information\n");
printf(" -s Show system information\n");
printf(" -c Show UCX configuration\n");
printf(" -C Comment-out default configuration values\n");
printf(" -a Show also hidden configuration\n");
printf(" -f Display fully decorated output\n");
printf("\nUCP information (-u is required):\n");
printf(" -p Show UCP context information\n");
printf(" -w Show UCP worker information\n");
printf(" -e Show UCP endpoint configuration\n");
printf(" -m <size> Show UCP memory allocation method for a given size\n");
printf(" -u <features> UCP context features to use. String of one or more of:\n");
printf(" 'a' : atomic operations\n");
printf(" 'r' : remote memory access\n");
printf(" 't' : tag matching \n");
printf(" 'm' : active messages \n");
printf(" Modifiers to use in combination with above features:\n");
printf(" 'w' : wakeup\n");
printf(" 'e' : error handling\n");
printf("\nOther settings:\n");
printf(" -t <name> Filter devices information using specified transport (requires -d)\n");
printf(" -n <count> Estimated UCP endpoint count (for ucp_init)\n");
printf(" -N <count> Estimated UCP endpoint count per node (for ucp_init)\n");
printf(" -D <type> Set which device types to use when creating UCP context:\n");
printf(" 'all' : all possible devices (default)\n");
printf(" 'shm' : shared memory devices only\n");
printf(" 'net' : network devices only\n");
printf(" 'self' : self transport only\n");
printf(" -P <type> Set peer process placement for printing UCP endpoint configuration:\n");
printf(" 'self' : same process (default)\n");
printf(" 'intra' : same node\n");
printf(" 'inter' : different node\n");
printf(" -A <ip> Local IP device address to use for creating\n"
" endpoint in client/server mode\n");
printf(" -6 IPv6 address specified with option -A\n");
printf(" -T Print system topology\n");
printf(" -M Print memory copy bandwidth\n");
printf(" -h Show this help message\n");
printf("\n");
}
static void ep_error_callback(void *arg, ucp_ep_h ep, ucs_status_t status)
{
}
int main(int argc, char **argv)
{
const uint64_t required_ucp_features = UCP_FEATURE_AMO32 |
UCP_FEATURE_AMO64 | UCP_FEATURE_RMA |
UCP_FEATURE_TAG | UCP_FEATURE_AM;
char *ip_addr = NULL;
sa_family_t ip_addr_family;
ucs_config_print_flags_t print_flags;
ucp_ep_params_t ucp_ep_params;
unsigned dev_type_bitmap;
process_placement_t proc_placement;
uint64_t ucp_features;
size_t ucp_num_eps;
size_t ucp_num_ppn;
unsigned print_opts;
char *tl_name, *mem_size;
const char *f;
int c;
print_opts = 0;
print_flags = (ucs_config_print_flags_t)0;
tl_name = NULL;
ucp_features = 0;
ucp_num_eps = 1;
ucp_num_ppn = 1;
mem_size = NULL;
dev_type_bitmap = UINT_MAX;
proc_placement = PROCESS_PLACEMENT_SELF;
ucp_ep_params.field_mask = 0;
ip_addr_family = AF_INET;
while ((c = getopt(argc, argv, "fahvc6ydbswpeCt:n:u:D:P:m:N:A:TM")) != -1) {
switch (c) {
case 'f':
print_flags |= UCS_CONFIG_PRINT_CONFIG | UCS_CONFIG_PRINT_HEADER | UCS_CONFIG_PRINT_DOC;
break;
case 'a':
print_flags |= UCS_CONFIG_PRINT_HIDDEN;
break;
case 'c':
print_flags |= UCS_CONFIG_PRINT_CONFIG;
break;
case 'C':
print_flags |= UCS_CONFIG_PRINT_COMMENT_DEFAULT;
break;
case 'v':
print_opts |= PRINT_VERSION;
break;
case 'd':
print_opts |= PRINT_DEVICES;
break;
case 'b':
print_opts |= PRINT_BUILD_CONFIG;
break;
case 'y':
print_opts |= PRINT_TYPES;
break;
case 's':
print_opts |= PRINT_SYS_INFO;
break;
case 'p':
print_opts |= PRINT_UCP_CONTEXT;
break;
case 'w':
print_opts |= PRINT_UCP_WORKER;
break;
case 'e':
print_opts |= PRINT_UCP_EP;
break;
case 'm':
print_opts |= PRINT_MEM_MAP;
mem_size = optarg;
break;
case 't':
tl_name = optarg;
break;
case 'n':
ucp_num_eps = atol(optarg);
break;
case 'N':
ucp_num_ppn = atol(optarg);
break;
case 'u':
for (f = optarg; *f; ++f) {
switch (*f) {
case 'a':
ucp_features |= UCP_FEATURE_AMO32|UCP_FEATURE_AMO64;
break;
case 'r':
ucp_features |= UCP_FEATURE_RMA;
break;
case 't':
ucp_features |= UCP_FEATURE_TAG;
break;
case 'm':
ucp_features |= UCP_FEATURE_AM;
break;
case 'w':
ucp_features |= UCP_FEATURE_WAKEUP;
break;
case 'e':
ucp_ep_params.field_mask |=
UCP_EP_PARAM_FIELD_ERR_HANDLING_MODE |
UCP_EP_PARAM_FIELD_ERR_HANDLER;
ucp_ep_params.err_mode = UCP_ERR_HANDLING_MODE_PEER;
ucp_ep_params.err_handler.cb = ep_error_callback;
break;
default:
usage();
return -1;
}
}
break;
case 'D':
if (!strcasecmp(optarg, "net")) {
dev_type_bitmap = UCS_BIT(UCT_DEVICE_TYPE_NET);
} else if (!strcasecmp(optarg, "shm")) {
dev_type_bitmap = UCS_BIT(UCT_DEVICE_TYPE_SHM);
} else if (!strcasecmp(optarg, "self")) {
dev_type_bitmap = (UCS_BIT(UCT_DEVICE_TYPE_SELF) |
UCS_BIT(UCT_DEVICE_TYPE_ACC));
} else if (strcasecmp(optarg, "all")) {
usage();
return -1;
}
break;
case 'P':
if (!strcasecmp(optarg, "intra")) {
proc_placement = PROCESS_PLACEMENT_INTRA;
} else if (!strcasecmp(optarg, "inter")) {
proc_placement = PROCESS_PLACEMENT_INTER;
} else if (strcasecmp(optarg, "self")) {
usage();
return -1;
}
break;
case 'A':
ip_addr = optarg;
break;
case '6':
ip_addr_family = AF_INET6;
break;
case 'T':
print_opts |= PRINT_SYS_TOPO;
break;
case 'M':
print_opts |= PRINT_MEMCPY_BW;
break;
case 'h':
usage();
return 0;
default:
usage();
return -1;
}
}
if ((print_opts == 0) && (print_flags == 0)) {
usage();
return -2;
}
if (print_opts & PRINT_VERSION) {
print_version();
}
if (print_opts & PRINT_BUILD_CONFIG) {
print_build_config();
}
if (print_opts & PRINT_TYPES) {
print_type_info(tl_name);
}
if ((print_opts & (PRINT_DEVICES | PRINT_SYS_TOPO)) ||
(print_flags & UCS_CONFIG_PRINT_CONFIG)) {
print_uct_info(print_opts, print_flags, tl_name);
}
if (print_opts & (PRINT_SYS_INFO | PRINT_MEMCPY_BW | PRINT_SYS_TOPO)) {
print_sys_info(print_opts);
}
if (print_flags & UCS_CONFIG_PRINT_CONFIG) {
ucs_config_parser_print_all_opts(stdout, UCS_DEFAULT_ENV_PREFIX,
print_flags, &ucs_config_global_list);
}
if (print_opts & (PRINT_UCP_CONTEXT|PRINT_UCP_WORKER|PRINT_UCP_EP|PRINT_MEM_MAP)) {
if (!(ucp_features & required_ucp_features)) {
printf("Please select at least one of 'a','r','t','m' UCP features "
"using -u switch.\n");
usage();
return -1;
}
return print_ucp_info(print_opts, print_flags, ucp_features,
&ucp_ep_params, ucp_num_eps, ucp_num_ppn,
dev_type_bitmap, proc_placement, mem_size,
ip_addr, ip_addr_family);
}
return 0;
}