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>
#ifdef STARNEIG_ENABLE_MPI
#include <mpi.h>
#endif



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

#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