1
1
Этот коммит содержится в:
Ralph Castain 2015-07-01 05:27:30 -05:00
родитель c8730b56e8
Коммит 6829e192ad
5 изменённых файлов: 0 добавлений и 270 удалений

Просмотреть файл

@ -1,34 +0,0 @@
#
# Copyright (c) 2015 Intel, Inc. All rights reserved
# $COPYRIGHT$
#
# Additional copyrights may follow
#
# $HEADER$
#
sources = \
rtc_omp.c \
rtc_omp.h \
rtc_omp_component.c
# Make the output library in this directory, and name it either
# mca_<type>_<name>.la (for DSO builds) or libmca_<type>_<name>.la
# (for static builds).
if MCA_BUILD_orte_rtc_omp_DSO
component_noinst =
component_install = mca_rtc_omp.la
else
component_noinst = libmca_rtc_omp.la
component_install =
endif
mcacomponentdir = $(ortelibdir)
mcacomponent_LTLIBRARIES = $(component_install)
mca_rtc_omp_la_SOURCES = $(sources)
mca_rtc_omp_la_LDFLAGS = -module -avoid-version
noinst_LTLIBRARIES = $(component_noinst)
libmca_rtc_omp_la_SOURCES =$(sources)
libmca_rtc_omp_la_LDFLAGS = -module -avoid-version

Просмотреть файл

@ -1,7 +0,0 @@
#
# owner/status file
# owner: institution that is responsible for this package
# status: e.g. active, maintenance, unmaintained
#
owner: INTEL
status: active

Просмотреть файл

@ -1,131 +0,0 @@
/*
* Copyright (c) 2015 Intel, Inc. All rights reserved
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "orte_config.h"
#include "orte/constants.h"
#include "orte/types.h"
#include <errno.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif /* HAVE_UNISTD_H */
#ifdef HAVE_STRING_H
#include <string.h>
#endif /* HAVE_STRING_H */
#include "opal/mca/hwloc/hwloc.h"
#include "opal/util/argv.h"
#include "opal/util/opal_environ.h"
#include "orte/util/show_help.h"
#include "orte/util/error_strings.h"
#include "orte/runtime/orte_globals.h"
#include "orte/mca/errmgr/errmgr.h"
#include "orte/mca/rmaps/rmaps_types.h"
#include "orte/mca/rtc/base/base.h"
#include "rtc_omp.h"
static int init(void);
static void finalize(void);
static void set(orte_job_t *jdata,
orte_proc_t *proc,
char ***environ_copy,
int write_fd);
orte_rtc_base_module_t orte_rtc_omp_module = {
init,
finalize,
NULL,
set,
NULL
};
static int init(void)
{
return ORTE_SUCCESS;
}
static void finalize(void)
{
return;
}
static void set(orte_job_t *jobdat,
orte_proc_t *child,
char ***environ_copy,
int write_fd)
{
char *param;
char *cpu_bitmap;
char **ranges, *ptr, *tmp, **results;
int i, n, start, end;
opal_output_verbose(2, orte_rtc_base_framework.framework_output,
"%s omp:set envars on child %s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
(NULL == child) ? "NULL" : ORTE_NAME_PRINT(&child->name));
if (NULL == jobdat || NULL == child) {
/* nothing for us to do */
opal_output_verbose(2, orte_rtc_base_framework.framework_output,
"%s omp:set jobdat %s child %s - nothing to do",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
(NULL == jobdat) ? "NULL" : ORTE_JOBID_PRINT(jobdat->jobid),
(NULL == child) ? "NULL" : ORTE_NAME_PRINT(&child->name));
return;
}
/* See if we are bound */
cpu_bitmap = NULL;
if (!orte_get_attribute(&child->attributes, ORTE_PROC_CPU_BITMAP, (void**)&cpu_bitmap, OPAL_STRING) ||
NULL == cpu_bitmap || 0 == strlen(cpu_bitmap)) {
/* we are not bound, so indicate that by setting OMP_PROC_BIND = false */
opal_output_verbose(2, orte_rtc_base_framework.framework_output,
"%s omp:set not bound - set OMP_PROC_BIND=FALSE",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME));
opal_setenv("OMP_PROC_BIND", "FALSE", true, environ_copy);
} else {
/* we are bound to something, so indicate that by setting OMP_PROC_BIND = true */
opal_output_verbose(2, orte_rtc_base_framework.framework_output,
"%s omp:set not bound - set OMP_PROC_BIND=SPREAD",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME));
opal_setenv("OMP_PROC_BIND", "SPREAD", true, environ_copy);
/* compose OMP_PLACES to indicate where we are bound - sadly, the OMP folks
* use a different syntax than HWLOC, an so we can't just provide the bitmap
* string. So we will traverse the bitmap and convert as required */
ranges = opal_argv_split(cpu_bitmap, ',');
results = NULL;
for (i=0; NULL != ranges[i]; i++) {
if (NULL == (ptr = strchr(ranges[i], '-'))) {
asprintf(&tmp, "{%s}", ranges[i]);
opal_argv_append_nosize(&results, tmp);
free(tmp);
} else {
*ptr = '\0';
++ptr;
start = strtol(ranges[i], NULL, 10);
end = strtol(ptr, NULL, 10);
for (n=start; n <= end; n++) {
asprintf(&tmp, "{%d}", n);
opal_argv_append_nosize(&results, tmp);
free(tmp);
}
}
}
opal_argv_free(ranges);
param = opal_argv_join(results, ',');
opal_argv_free(results);
opal_setenv("OMP_PLACES", param, true, environ_copy);
free(param);
}
if (NULL != cpu_bitmap) {
free(cpu_bitmap);
}
}

Просмотреть файл

@ -1,28 +0,0 @@
/*
* Copyright (c) 2015 Intel, Inc. All rights reserved
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
/**
* @file
*/
#ifndef ORTE_RTC_OMP_H
#define ORTE_RTC_OMP_H
#include "orte_config.h"
#include "orte/mca/rtc/rtc.h"
BEGIN_C_DECLS
ORTE_MODULE_DECLSPEC extern orte_rtc_base_component_t mca_rtc_omp_component;
extern orte_rtc_base_module_t orte_rtc_omp_module;
END_C_DECLS
#endif /* ORTE_RTC_OMP_H */

Просмотреть файл

@ -1,70 +0,0 @@
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
/*
* Copyright (c) 2015 Intel, Inc. All rights reserved
* Copyright (c) 2015 Los Alamos National Security, LLC. All rights
* reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "orte_config.h"
#include "orte/constants.h"
#include "opal/mca/base/base.h"
#include "rtc_omp.h"
/*
* Local functions
*/
static int rtc_omp_query(mca_base_module_t **module, int *priority);
static int rtc_omp_register(void);
static int my_priority;
orte_rtc_base_component_t mca_rtc_omp_component = {
.base_version = {
ORTE_RTC_BASE_VERSION_1_0_0,
.mca_component_name = "omp",
MCA_BASE_MAKE_VERSION(component, ORTE_MAJOR_VERSION, ORTE_MINOR_VERSION,
ORTE_RELEASE_VERSION),
.mca_query_component = rtc_omp_query,
.mca_register_component_params = rtc_omp_register,
},
.base_data = {
/* The component is checkpoint ready */
MCA_BASE_METADATA_PARAM_CHECKPOINT
},
};
static int rtc_omp_register(void)
{
mca_base_component_t *c = &mca_rtc_omp_component.base_version;
/* set below the default */
my_priority = 50;
(void) mca_base_component_var_register (c, "priority", "Priority of the OMP rtc component",
MCA_BASE_VAR_TYPE_INT, NULL, 0, 0,
OPAL_INFO_LVL_9,
MCA_BASE_VAR_SCOPE_READONLY,
&my_priority);
return ORTE_SUCCESS;
}
static int rtc_omp_query(mca_base_module_t **module, int *priority)
{
/* Only run on the HNP */
*priority = my_priority;
*module = (mca_base_module_t *)&orte_rtc_omp_module;
return ORTE_SUCCESS;
}