numThreads.h

Go to the documentation of this file.

/*********************************************************************************/
/*  Copyright 2009-2024 Barcelona Supercomputing Center                          */
/*                                                                               */
/*  This file is part of the DLB library.                                        */
/*                                                                               */
/*  DLB is free software: you can redistribute it and/or modify                  */
/*  it under the terms of the GNU Lesser General Public License as published by  */
/*  the Free Software Foundation, either version 3 of the License, or            */
/*  (at your option) any later version.                                          */
/*                                                                               */
/*  DLB is distributed in the hope that it will be useful,                       */
/*  but WITHOUT ANY WARRANTY; without even the implied warranty of               */
/*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the                */
/*  GNU Lesser General Public License for more details.                          */
/*                                                                               */
/*  You should have received a copy of the GNU Lesser General Public License     */
/*  along with DLB.  If not, see <https://www.gnu.org/licenses/>.                */
/*********************************************************************************/
 
#ifndef NUMTHREADS_H
#define NUMTHREADS_H
 
#include "apis/dlb_types.h"
 
#include <sched.h>
#include <stdbool.h>
 
typedef struct pm_interface {
    /* Callbacks list */
    dlb_callback_set_num_threads_t   dlb_callback_set_num_threads_ptr;
    void                            *dlb_callback_set_num_threads_arg;
    dlb_callback_set_active_mask_t   dlb_callback_set_active_mask_ptr;
    void                            *dlb_callback_set_active_mask_arg;
    dlb_callback_set_process_mask_t  dlb_callback_set_process_mask_ptr;
    void                            *dlb_callback_set_process_mask_arg;
    dlb_callback_add_active_mask_t   dlb_callback_add_active_mask_ptr;
    void                            *dlb_callback_add_active_mask_arg;
    dlb_callback_add_process_mask_t  dlb_callback_add_process_mask_ptr;
    void                            *dlb_callback_add_process_mask_arg;
    dlb_callback_enable_cpu_t        dlb_callback_enable_cpu_ptr;
    void                            *dlb_callback_enable_cpu_arg;
    dlb_callback_disable_cpu_t       dlb_callback_disable_cpu_ptr;
    void                            *dlb_callback_disable_cpu_arg;
    dlb_callback_enable_cpu_set_t    dlb_callback_enable_cpu_set_ptr;
    void                            *dlb_callback_enable_cpu_set_arg;
    dlb_callback_disable_cpu_set_t   dlb_callback_disable_cpu_set_ptr;
    void                            *dlb_callback_disable_cpu_set_arg;
} pm_interface_t;
 
void pm_init(pm_interface_t *pm);
void pm_finalize(pm_interface_t *pm);
int pm_get_num_threads(void);
int pm_callback_set(pm_interface_t *pm, dlb_callbacks_t which,
        dlb_callback_t callback, void *arg);
int pm_callback_get(const pm_interface_t *pm, dlb_callbacks_t which,
        dlb_callback_t *callback, void **arg);
 
int update_threads(const pm_interface_t *pm, int threads);
int set_mask(const pm_interface_t *pm, const cpu_set_t *cpu_set);
int set_process_mask(const pm_interface_t *pm, const cpu_set_t *cpu_set);
int add_mask(const pm_interface_t *pm, const cpu_set_t *cpu_set);
int add_process_mask(const pm_interface_t *pm, const cpu_set_t *cpu_set);
int enable_cpu(const pm_interface_t *pm, int cpuid);
int disable_cpu(const pm_interface_t *pm, int cpuid);
int enable_cpu_set(const pm_interface_t *pm, const cpu_set_t *cpu_set);
int disable_cpu_set(const pm_interface_t *pm, const cpu_set_t *cpu_set);
 
#endif //NUMTHREADS_H