node.h

Go to the documentation of this file.

 
#ifndef STARNEIG_NODE_H
#define STARNEIG_NODE_H
 
#ifdef __cplusplus
extern "C" {
#endif
 
#include <starneig/configuration.h>
#include <stddef.h>
#ifdef STARNEIG_ENABLE_MPI
#include <mpi.h>
#endif
 
 
 
#define STARNEIG_USE_ALL                -1
 
typedef unsigned starneig_flag_t;
 
#define STARNEIG_DEFAULT                0x0
 
#define STARNEIG_HINT_SM                0x0
 
#define STARNEIG_HINT_DM                0x1
 
#define STARNEIG_FXT_DISABLE            0x2
 
#define STARNEIG_AWAKE_WORKERS          0x4
 
#define STARNEIG_AWAKE_MPI_WORKER       0x8
 
#define STARNEIG_FAST_DM (STARNEIG_HINT_DM | STARNEIG_AWAKE_WORKERS | STARNEIG_AWAKE_MPI_WORKER)
 
#define STARNEIG_NO_VERBOSE             0x10
 
#define STARNEIG_NO_MESSAGES            (STARNEIG_NO_VERBOSE | 0x20)
 
 
void starneig_node_init(int cores, int gpus, starneig_flag_t flags);
 
int starneig_node_initialized();
 
int starneig_node_get_cores();
 
void starneig_node_set_cores(int cores);
 
int starneig_node_get_gpus();
 
void starneig_node_set_gpus(int gpus);
 
void starneig_node_finalize();
 
#ifdef STARNEIG_ENABLE_CUDA
 
 
void starneig_node_enable_pinning();
 
void starneig_node_disable_pinning();
 
 
#endif
 
 
// deprecated
#ifdef STARNEIG_ENABLE_MPI
void starneig_mpi_set_comm(MPI_Comm comm);
MPI_Comm starneig_mpi_get_comm();
#endif
 
#ifdef __cplusplus
}
#endif
 
#endif // STARNEIG_NODE_H