nm_launcher_private.h

Go to the documentation of this file.

/*
 * NewMadeleine
 * Copyright (C) 2006-2026 (see AUTHORS file)
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or (at
 * your option) any later version.
 *
 * This program 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
 * General Public License for more details.
 */
 
#ifndef NM_LAUNCHER_PRIVATE_H
#define NM_LAUNCHER_PRIVATE_H
 
#include <nm_launcher_interface.h>
#include <nm_core_interface.h>
#include <Padico/Puk.h>
 
struct nm_launcher_info_s
{
  int size; 
  int rank; 
  int wide_url_support; 
};
 
struct nm_launcher_driver_s
{
  void (*init)(void*_status, int*argc, char**argv, const char*label);
  void (*barrier)(void*_status);
  void (*get_gates)(void*_status, nm_gate_t*gates);
  void (*get_topology)(void*_status, struct nm_launcher_rank_vect_s*p_ranks,
                       struct nm_launcher_subgroup_vect_s*p_subgroups, int*n_subgroups);
  void (*abort)(void*_status, int rc);
};
 
PUK_IFACE_TYPE(NewMad_Launcher, struct nm_launcher_driver_s);
 
struct nm_launcher_gate_s
{
  nm_gate_t p_gate;
  char*p_url;
  struct nm_drv_vect_s drvs;
};
 
struct nm_launcher_gates_s
{
  int size;
  int rank;
  struct nm_launcher_gate_s*p_gates;
};
 
PUK_HASHTABLE_TYPE(nm_component, puk_component_t, nm_drv_t,
                   &puk_hash_pointer_default_hash, &puk_hash_pointer_default_eq, NULL);
 
void nm_launcher_gates_init(struct nm_launcher_gates_s*p_gates, int size, int rank);
 
void nm_launcher_gates_destroy(struct nm_launcher_gates_s*p_gates);
 
void nm_launcher_gates_connect(struct nm_launcher_gates_s*p_gates);
 
 
/* ********************************************************* */
/* ** interface for launchers to interact with nmad */
 
void nm_launcher_set_info(const struct nm_launcher_info_s*p_info);
 
nm_drv_t nm_launcher_add_driver(puk_component_t component, nm_trk_kind_t kind);
 
typedef nm_drv_vect_t (*nm_launcher_selector_t)(const char*url, void*_arg);
void nm_launcher_set_selector(nm_launcher_selector_t selector, void*_arg);
 
int nm_launcher_get_url(const char**p_local_url);
 
int nm_launcher_connect_async(nm_gate_t*pp_gate, const char*remote_url);
 
int nm_launcher_connect_wait(nm_gate_t p_gate);
 
 
/* ** direct send/recv functions, without session */
 
void nm_launcher_isend(struct nm_req_s*p_req, struct nm_data_s*p_data, struct nm_gate_s*p_gate);
 
void nm_launcher_send(struct nm_data_s*p_data, struct nm_gate_s*p_gate);
 
void nm_launcher_irecv(struct nm_req_s*p_req, struct nm_data_s*p_data, struct nm_gate_s*p_gate);
 
void nm_launcher_recv(struct nm_data_s*p_data, struct nm_gate_s*p_gate);
 
#endif /* NM_LAUNCHER_PRIVATE_H */