159 строки
4.6 KiB
C
159 строки
4.6 KiB
C
/*
|
|
* Copyright (c) 2004-2008 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2005 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2004-2005 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.
|
|
* Copyright (c) 2007 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2008 Sun Microsystems, Inc. All rights reserved.
|
|
*
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "opal_config.h"
|
|
#include "opal/constants.h"
|
|
|
|
#include "opal/mca/base/mca_base_param.h"
|
|
#include "opal/mca/paffinity/paffinity.h"
|
|
#include "opal/mca/paffinity/base/base.h"
|
|
|
|
#include "paffinity_test.h"
|
|
|
|
/*
|
|
* Local functions
|
|
*/
|
|
static int init(void);
|
|
static int set(opal_paffinity_base_cpu_set_t cpumask);
|
|
static int get(opal_paffinity_base_cpu_set_t *cpumask);
|
|
static int finalize(void);
|
|
static int map_to_processor_id(int socket, int core, int *processor_id);
|
|
static int map_to_socket_core(int processor_id, int *socket, int *core);
|
|
static int get_processor_info(int *num_processors);
|
|
static int get_socket_info(int *num_sockets);
|
|
static int get_core_info(int socket, int *num_cores);
|
|
static int get_physical_processor_id(int logical_processor_id);
|
|
static int get_physical_socket_id(int logical_socket_id);
|
|
static int get_physical_core_id(int physical_socket_id, int logical_core_id);
|
|
|
|
/*
|
|
* Test paffinity module
|
|
*/
|
|
opal_paffinity_base_module_t opal_paffinity_test_module = {
|
|
/* Initialization function */
|
|
init,
|
|
|
|
/* Module function pointers */
|
|
set,
|
|
get,
|
|
map_to_processor_id,
|
|
map_to_socket_core,
|
|
get_processor_info,
|
|
get_socket_info,
|
|
get_core_info,
|
|
get_physical_processor_id,
|
|
get_physical_socket_id,
|
|
get_physical_core_id,
|
|
finalize
|
|
};
|
|
|
|
/* nothing to init here */
|
|
static int init(void)
|
|
{
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
/* this gives us a cpumask which tells which CPU to bind */
|
|
static int set(opal_paffinity_base_cpu_set_t cpumask)
|
|
{
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
/* This get function returns the CPU id that's currently binded,
|
|
* and then sets the cpumask. */
|
|
static int get(opal_paffinity_base_cpu_set_t *cpumask)
|
|
{
|
|
int i;
|
|
|
|
OPAL_PAFFINITY_CPU_ZERO(*cpumask);
|
|
if (mca_paffinity_test_component.bound) {
|
|
for (i=0; i < mca_paffinity_test_component.num_sockets*mca_paffinity_test_component.num_cores; i+=2) {
|
|
OPAL_PAFFINITY_CPU_SET(i, *cpumask);
|
|
}
|
|
/* assign all cores in the 2nd socket, if it exists */
|
|
if (mca_paffinity_test_component.num_sockets >= 2) {
|
|
for (i=mca_paffinity_test_component.num_cores; i < 2*mca_paffinity_test_component.num_cores; i++) {
|
|
OPAL_PAFFINITY_CPU_SET(i, *cpumask);
|
|
}
|
|
}
|
|
} else {
|
|
for (i=0; i < mca_paffinity_test_component.num_sockets*mca_paffinity_test_component.num_cores; i++) {
|
|
OPAL_PAFFINITY_CPU_SET(i, *cpumask);
|
|
}
|
|
}
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
static int map_to_processor_id(int socket, int core, int *processor_id)
|
|
{
|
|
*processor_id = socket*mca_paffinity_test_component.num_cores + core;
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
static int map_to_socket_core(int processor_id, int *socket, int *core)
|
|
{
|
|
*socket = processor_id / mca_paffinity_test_component.num_cores;
|
|
*core = processor_id % mca_paffinity_test_component.num_cores;
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
static int get_processor_info(int *num_processors)
|
|
{
|
|
*num_processors = mca_paffinity_test_component.num_sockets * mca_paffinity_test_component.num_cores;
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
static int get_socket_info(int *num_sockets)
|
|
{
|
|
*num_sockets = mca_paffinity_test_component.num_sockets;
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
static int get_core_info(int socket, int *num_cores)
|
|
{
|
|
*num_cores = mca_paffinity_test_component.num_cores;
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
static int get_physical_processor_id(int logical_processor_id)
|
|
{
|
|
return logical_processor_id;
|
|
}
|
|
|
|
static int get_physical_socket_id(int logical_socket_id)
|
|
{
|
|
return logical_socket_id;
|
|
}
|
|
|
|
static int get_physical_core_id(int physical_socket_id, int logical_core_id)
|
|
{
|
|
if (mca_paffinity_test_component.num_cores < logical_core_id) {
|
|
return OPAL_ERROR;
|
|
}
|
|
return logical_core_id;
|
|
}
|
|
|
|
static int finalize(void)
|
|
{
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|