spd.h

Go to the documentation of this file.

/*********************************************************************************/
/*  Copyright 2009-2021 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 SPD_H
#define SPD_H
 
#include "LB_core/lb_funcs.h"
 
#include "LB_numThreads/numThreads.h"
#include "support/options.h"
#include "support/types.h"
 
#include <sys/types.h>
 
/* Sub-process Descriptor */
 
typedef struct SubProcessDescriptor {
    pid_t id;   // sub-proces id, used everywhere to identify spd
    pid_t pid;  // real pid for fork-safety purposes
    bool dlb_initialized;
    bool dlb_preinitialized;
    bool dlb_initialized_via_ompt;
    bool lewi_enabled;
    cpu_set_t process_mask;
    cpu_set_t active_mask;
    options_t options;
    pm_interface_t pm;
    policy_t lb_policy;
    balance_policy_t lb_funcs;
    unsigned int lewi_ncpus;
    void *lewi_info;
    void *talp_info;
    void *barrier_info;
} subprocess_descriptor_t;
 
extern __thread subprocess_descriptor_t *thread_spd;
 
void spd_enter_dlb(subprocess_descriptor_t *spd);
void spd_register(subprocess_descriptor_t *spd);
void spd_unregister(const subprocess_descriptor_t *spd);
void spd_set_pthread(const subprocess_descriptor_t *spd, pthread_t pthread);
pthread_t spd_get_pthread(const subprocess_descriptor_t *spd);
const subprocess_descriptor_t** spd_get_spds(void);
 
#endif /* SPD_H */