1
1

Update the Windows paffinity component.

This commit was SVN r15027.
Этот коммит содержится в:
George Bosilca 2007-06-12 22:48:33 +00:00
родитель 715f6012cf
Коммит c2a18b0741
4 изменённых файлов: 21 добавлений и 18 удалений

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

@ -25,7 +25,6 @@
#include "opal/mca/paffinity/paffinity.h"
/*
* Global functions for MCA overall paffinity open and close
*/

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

@ -52,14 +52,14 @@ extern "C" {
/**
* Globally exported variable
*/
OPAL_MODULE_DECLSPEC extern const opal_paffinity_base_component_1_0_0_t
OPAL_MODULE_DECLSPEC extern const opal_paffinity_base_component_1_1_0_t
mca_paffinity_windows_component;
/**
* paffinity query API function
*/
const opal_paffinity_base_module_1_0_0_t *
const opal_paffinity_base_module_1_1_0_t *
opal_paffinity_windows_component_query(int *query);
#if defined(c_plusplus) || defined(__cplusplus)

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

@ -44,7 +44,7 @@ static int windows_open(void);
* and pointers to our public functions in it
*/
const opal_paffinity_base_component_1_0_0_t mca_paffinity_windows_component = {
const opal_paffinity_base_component_1_1_0_t mca_paffinity_windows_component = {
/* First, the mca_component_t struct containing meta information
about the component itself */
@ -53,7 +53,7 @@ const opal_paffinity_base_component_1_0_0_t mca_paffinity_windows_component = {
/* Indicate that we are a paffinity v1.0.0 component (which also
implies a specific MCA version) */
OPAL_PAFFINITY_BASE_VERSION_1_0_0,
OPAL_PAFFINITY_BASE_VERSION_1_1_0,
/* Component name and version */

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

@ -28,30 +28,30 @@
* Local functions
*/
static int windows_module_init(void);
static int windows_module_finalize(void);
static int windows_module_get_num_procs(int *num_procs);
static int windows_module_set(int id);
static int windows_module_get(int *id);
static int windows_module_set(opal_paffinity_base_cpu_set_t cpumask);
static int windows_module_get(opal_paffinity_base_cpu_set_t *cpumask);
static SYSTEM_INFO sys_info;
/*
* Linux paffinity module
*/
static const opal_paffinity_base_module_1_0_0_t module = {
static const opal_paffinity_base_module_1_1_0_t module = {
/* Initialization function */
windows_module_init,
/* Module function pointers */
windows_module_get_num_procs,
windows_module_set,
windows_module_get
windows_module_get,
windows_module_finalize
};
const opal_paffinity_base_module_1_0_0_t *
const opal_paffinity_base_module_1_1_0_t *
opal_paffinity_windows_component_query(int *query)
{
int param;
@ -62,6 +62,10 @@ opal_paffinity_windows_component_query(int *query)
return &module;
}
static int windows_module_finalize(void)
{
return OPAL_SUCCESS;
}
static int windows_module_init(void)
{
@ -77,7 +81,7 @@ static int windows_module_get_num_procs(int *num_procs)
return OPAL_SUCCESS;
}
static int windows_module_set(int id)
static int windows_module_set(opal_paffinity_base_cpu_set_t cpumask)
{
HANDLE threadid = GetCurrentThread();
DWORD_PTR process_mask, system_mask;
@ -86,8 +90,8 @@ static int windows_module_set(int id)
return OPAL_ERR_NOT_FOUND;
}
if( (int)(1 << id) & (system_mask & 0xFFFFFFFF) ) {
process_mask = (int)(1 << id);
if( (int)(1 << cpumask.bitmask[0]) & (system_mask & 0xFFFFFFFF) ) {
process_mask = (int)(1 << cpumask.bitmask[0]);
if( SetThreadAffinityMask( threadid, process_mask ) )
return OPAL_SUCCESS;
/* otherwise something went wrong */
@ -97,16 +101,16 @@ static int windows_module_set(int id)
}
static int windows_module_get(int *id)
static int windows_module_get(opal_paffinity_base_cpu_set_t *cpumask)
{
HANDLE threadid = GetCurrentThread();
DWORD_PTR process_mask, system_mask;
if( GetProcessAffinityMask( threadid, &process_mask, &system_mask ) ) {
*id = (int)process_mask;
cpumask->bitmask[0] = (opal_paffinity_base_bitmask_t)process_mask;
return OPAL_SUCCESS;
}
*id = 1;
cpumask->bitmask[0] = 1;
return OPAL_ERR_BAD_PARAM;
}