StarPU Internal Handbook
Loading...
Searching...
No Matches
driver_mpi_source.h
Go to the documentation of this file.
1/* StarPU --- Runtime system for heterogeneous multicore architectures.
2 *
3 * Copyright (C) 2016-2023 Université de Bordeaux, CNRS (LaBRI UMR 5800), Inria
4 *
5 * StarPU is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU Lesser General Public License as published by
7 * the Free Software Foundation; either version 2.1 of the License, or (at
8 * your option) any later version.
9 *
10 * StarPU is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
13 *
14 * See the GNU Lesser General Public License in COPYING.LGPL for more details.
15 */
16
17#ifndef __DRIVER_MPI_SOURCE_H__
18#define __DRIVER_MPI_SOURCE_H__
19
22#include <core/workers.h>
24#include <datawizard/node_ops.h>
25
26#pragma GCC visibility push(hidden)
27
28void _starpu_mpi_ms_preinit(void);
29
30#ifdef STARPU_USE_MPI_MASTER_SLAVE
31extern struct _starpu_node_ops _starpu_driver_mpi_ms_node_ops;
32
36
37unsigned _starpu_mpi_src_get_device_count();
38void *_starpu_mpi_src_worker(void *arg);
39
40void _starpu_init_mpi_config(struct _starpu_machine_topology *topology, struct _starpu_machine_config *config,
41 struct starpu_conf *user_conf, int no_mp_config);
42void _starpu_mpi_init_worker_binding(struct _starpu_machine_config *config, int no_mp_config STARPU_ATTRIBUTE_UNUSED, struct _starpu_worker *workerarg);
43void _starpu_mpi_init_worker_memory(struct _starpu_machine_config *config, int no_mp_config STARPU_ATTRIBUTE_UNUSED, struct _starpu_worker *workerarg);
44void _starpu_deinit_mpi_config(struct _starpu_machine_config *config);
45
46void _starpu_mpi_source_init(struct _starpu_mp_node *node);
47void _starpu_mpi_source_deinit(struct _starpu_mp_node *node);
48
49int _starpu_mpi_is_direct_access_supported(unsigned node, unsigned handling_node);
50
51#endif /* STARPU_USE_MPI_MASTER_SLAVE */
52
53#pragma GCC visibility pop
54
55#endif /* __DRIVER_MPI_SOURCE_H__ */
struct _starpu_mp_node * _starpu_mpi_ms_src_get_actual_thread_mp_node()
Definition workers.h:441
Definition workers.h:352
Definition node_ops.h:92
Definition workers.h:155