1
1
openmpi/ompi/tools/ompi_info/ompi_info.cc
Ralph Castain 0532d799d6 Complete implementation of the --without-rte-support configure option. Working with Brian, this has been tested on RedStorm.
Some minor changes to help facilitate debugger support so that both mpirun and yod can operate with it. Still to be completed.

This commit was SVN r18664.
2008-06-18 03:15:56 +00:00

299 строки
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-2006 The University of Tennessee and The University
// of Tennessee Research Foundation. All rights
// reserved.
// Copyright (c) 2004-2007 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.
// Copryight (c) 2007-2008 Cisco Systems, Inc. All rights reserved.
// $COPYRIGHT$
//
// Additional copyrights may follow
//
// $HEADER$
//
#include "ompi_config.h"
#include <iostream>
#include <string>
#include <utility>
#include <list>
#include <stdio.h>
#include <ctype.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef HAVE_NETDB_H
#include <netdb.h>
#endif
#ifdef HAVE_SYS_PARAM_H
#include <sys/param.h>
#endif
#include <errno.h>
#include "opal/mca/installdirs/installdirs.h"
#include "opal/class/opal_object.h"
#include "opal/runtime/opal.h"
#include "orte/runtime/runtime.h"
#include "orte/util/show_help.h"
#include "opal/util/cmd_line.h"
#include "opal/util/argv.h"
#include "ompi/communicator/communicator.h"
#include "opal/mca/base/base.h"
#include "ompi/tools/ompi_info/ompi_info.h"
#include "opal/threads/mutex.h"
using namespace std;
using namespace ompi_info;
//
// Public variables
//
bool ompi_info::pretty = true;
opal_cmd_line_t *ompi_info::cmd_line = NULL;
const string ompi_info::type_all = "all";
const string ompi_info::type_ompi = "ompi";
const string ompi_info::type_orte = "orte";
const string ompi_info::type_opal = "opal";
const string ompi_info::type_base = "base";
ompi_info::type_vector_t ompi_info::mca_types;
int main(int argc, char *argv[])
{
int ret = 0;
bool want_help = false;
bool cmd_error = false;
bool acted = false;
bool want_all = false;
char **app_env = NULL, **global_env = NULL;
int i, len;
// Initialize the argv parsing handle
if (OMPI_SUCCESS != opal_init_util()) {
orte_show_help("help-ompi_info.txt", "lib-call-fail", true,
"opal_init_util", __FILE__, __LINE__, NULL);
exit(ret);
}
cmd_line = OBJ_NEW(opal_cmd_line_t);
if (NULL == cmd_line) {
ret = errno;
orte_show_help("help-ompi_info.txt", "lib-call-fail", true,
"opal_cmd_line_create", __FILE__, __LINE__, NULL);
exit(ret);
}
opal_cmd_line_make_opt3(cmd_line, 'v', NULL, "version", 2,
"Show version of Open MPI or a component. The first parameter can be the keywords \"ompi\" or \"all\", a framework name (indicating all components in a framework), or a framework:component string (indicating a specific component). The second parameter can be one of: full, major, minor, release, greek, svn.");
opal_cmd_line_make_opt3(cmd_line, '\0', NULL, "param", 2,
"Show MCA parameters. The first parameter is the framework (or the keyword \"all\"); the second parameter is the specific component name (or the keyword \"all\").");
opal_cmd_line_make_opt3(cmd_line, '\0', NULL, "internal", 0,
"Show internal MCA parameters (not meant to be modified by users)");
opal_cmd_line_make_opt3(cmd_line, '\0', NULL, "path", 1,
"Show paths that Open MPI was configured with. Accepts the following parameters: prefix, bindir, libdir, incdir, mandir, pkglibdir, sysconfdir");
opal_cmd_line_make_opt3(cmd_line, '\0', NULL, "arch", 0,
"Show architecture Open MPI was compiled on");
opal_cmd_line_make_opt3(cmd_line, 'c', NULL, "config", 0,
"Show configuration options");
opal_cmd_line_make_opt3(cmd_line, 'h', NULL, "help", 0,
"Show this help message");
opal_cmd_line_make_opt3(cmd_line, '\0', NULL, "pretty", 0,
"When used in conjunction with other parameters, the output is displayed in 'prettyprint' format (default)");
opal_cmd_line_make_opt3(cmd_line, '\0', NULL, "parsable", 0,
"When used in conjunction with other parameters, the output is displayed in a machine-parsable format");
opal_cmd_line_make_opt3(cmd_line, '\0', NULL, "parseable", 0,
"Synonym for --parsable");
opal_cmd_line_make_opt3(cmd_line, '\0', NULL, "hostname", 0,
"Show the hostname that Open MPI was configured "
"and built on");
opal_cmd_line_make_opt3(cmd_line, 'a', NULL, "all", 0,
"Show all configuration options and MCA parameters");
// Call some useless functions in order to guarantee to link in some
// global variables. Only check the return value so that the
// compiler doesn't optimize out the useless function.
if (OMPI_SUCCESS != ompi_comm_link_function()) {
// Stop .. or I'll say stop again!
++ret;
} else {
--ret;
}
// set our threading level
opal_set_using_threads(false);
// Get MCA parameters, if any */
if( OMPI_SUCCESS != mca_base_open() ) {
orte_show_help("help-ompi_info.txt", "lib-call-fail", true, "mca_base_open", __FILE__, __LINE__ );
exit(1);
}
mca_base_cmd_line_setup(cmd_line);
// Do the parsing
if (OMPI_SUCCESS != opal_cmd_line_parse(cmd_line, false, argc, argv)) {
cmd_error = true;
}
if (!cmd_error &&
(opal_cmd_line_is_taken(cmd_line, "help") ||
opal_cmd_line_is_taken(cmd_line, "h"))) {
want_help = true;
}
if (cmd_error || want_help) {
char *usage = opal_cmd_line_get_usage_msg(cmd_line);
orte_show_help("help-ompi_info.txt", "usage", true, usage);
free(usage);
exit(cmd_error ? 1 : 0);
}
mca_base_cmd_line_process_args(cmd_line, &app_env, &global_env);
// putenv() all the stuff that we got back from env (in case the
// user specified some --mca params on the command line). This
// creates a memory leak, but that's unfortunately how putenv()
// works. :-(
len = opal_argv_count(app_env);
for (i = 0; i < len; ++i) {
putenv(app_env[i]);
}
len = opal_argv_count(global_env);
for (i = 0; i < len; ++i) {
putenv(global_env[i]);
}
ompi_info::mca_types.push_back("mca");
ompi_info::mca_types.push_back("mpi");
ompi_info::mca_types.push_back("orte");
ompi_info::mca_types.push_back("opal");
ompi_info::mca_types.push_back("filter");
ompi_info::mca_types.push_back("backtrace");
ompi_info::mca_types.push_back("memchecker");
ompi_info::mca_types.push_back("memory");
ompi_info::mca_types.push_back("paffinity");
ompi_info::mca_types.push_back("carto");
ompi_info::mca_types.push_back("maffinity");
ompi_info::mca_types.push_back("timer");
ompi_info::mca_types.push_back("installdirs");
#if OPAL_ENABLE_FT == 1
ompi_info::mca_types.push_back("crs");
#endif
ompi_info::mca_types.push_back("dpm");
ompi_info::mca_types.push_back("pubsub");
ompi_info::mca_types.push_back("allocator");
ompi_info::mca_types.push_back("coll");
ompi_info::mca_types.push_back("io");
ompi_info::mca_types.push_back("mpool");
ompi_info::mca_types.push_back("pml");
ompi_info::mca_types.push_back("bml");
ompi_info::mca_types.push_back("rcache");
ompi_info::mca_types.push_back("btl");
ompi_info::mca_types.push_back("mtl");
ompi_info::mca_types.push_back("topo");
ompi_info::mca_types.push_back("osc");
ompi_info::mca_types.push_back("common");
#if OPAL_ENABLE_FT == 1
ompi_info::mca_types.push_back("crcp");
#endif
#if !ORTE_DISABLE_FULL_SUPPORT
ompi_info::mca_types.push_back("iof");
ompi_info::mca_types.push_back("oob");
ompi_info::mca_types.push_back("odls");
ompi_info::mca_types.push_back("ras");
ompi_info::mca_types.push_back("rmaps");
ompi_info::mca_types.push_back("rml");
ompi_info::mca_types.push_back("routed");
ompi_info::mca_types.push_back("plm");
#if OPAL_ENABLE_FT == 1
ompi_info::mca_types.push_back("snapc");
#endif
ompi_info::mca_types.push_back("filem");
#endif
/* these are always included */
ompi_info::mca_types.push_back("errmgr");
ompi_info::mca_types.push_back("ess");
ompi_info::mca_types.push_back("grpcomm");
// Execute the desired action(s)
if (opal_cmd_line_is_taken(cmd_line, "pretty")) {
ompi_info::pretty = true;
} else if (opal_cmd_line_is_taken(cmd_line, "parsable") || opal_cmd_line_is_taken(cmd_line, "parseable")) {
ompi_info::pretty = false;
}
want_all = opal_cmd_line_is_taken(cmd_line, "all");
if (want_all || opal_cmd_line_is_taken(cmd_line, "version")) {
do_version(want_all, cmd_line);
acted = true;
}
if (want_all || opal_cmd_line_is_taken(cmd_line, "path")) {
do_path(want_all, cmd_line);
acted = true;
}
if (want_all || opal_cmd_line_is_taken(cmd_line, "arch")) {
do_arch();
acted = true;
}
if (want_all || opal_cmd_line_is_taken(cmd_line, "hostname")) {
do_hostname();
acted = true;
}
if (want_all || opal_cmd_line_is_taken(cmd_line, "config")) {
do_config(true);
acted = true;
}
if (want_all || opal_cmd_line_is_taken(cmd_line, "param")) {
do_params(want_all, opal_cmd_line_is_taken(cmd_line, "internal"));
acted = true;
}
// If no command line args are specified, show default set
if (!acted) {
ompi_info::show_ompi_version(ver_full);
ompi_info::show_path(path_prefix, opal_install_dirs.prefix);
ompi_info::do_arch();
ompi_info::do_hostname();
ompi_info::do_config(false);
ompi_info::open_components();
for (ompi_info::type_vector_t::size_type i = 0;
i < mca_types.size(); ++i) {
if ("mpi" != mca_types[i]) {
ompi_info::show_component_version(mca_types[i], component_all,
ver_full, type_all);
}
}
}
// All done
if (NULL != app_env) {
opal_argv_free(app_env);
}
if (NULL != global_env) {
opal_argv_free(global_env);
}
ompi_info::close_components();
OBJ_RELEASE(cmd_line);
mca_base_close();
opal_finalize_util();
return 0;
}