StarPU Internal Handbook
driver_mpi_source.h
Go to the documentation of this file.
1 /* StarPU --- Runtime system for heterogeneous multicore architectures.
2  *
3  * Copyright (C) 2016-2022 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 
28 void _starpu_mpi_ms_preinit(void);
29 
30 #ifdef STARPU_USE_MPI_MASTER_SLAVE
31 extern struct _starpu_node_ops _starpu_driver_mpi_ms_node_ops;
32 
36 
37 unsigned _starpu_mpi_src_get_device_count();
38 void *_starpu_mpi_src_worker(void *arg);
39 
40 void _starpu_init_mpi_config(struct _starpu_machine_topology *topology, struct _starpu_machine_config *config,
41  struct starpu_conf *user_conf, int no_mp_config);
42 void _starpu_mpi_init_worker_binding(struct _starpu_machine_config *config, int no_mp_config STARPU_ATTRIBUTE_UNUSED, struct _starpu_worker *workerarg);
43 void _starpu_mpi_init_worker_memory(struct _starpu_machine_config *config, int no_mp_config STARPU_ATTRIBUTE_UNUSED, struct _starpu_worker *workerarg);
44 void _starpu_deinit_mpi_config(struct _starpu_machine_config *config);
45 
46 void _starpu_mpi_source_init(struct _starpu_mp_node *node);
47 void _starpu_mpi_source_deinit(struct _starpu_mp_node *node);
48 
49 int _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