1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
/* StarPU --- Runtime system for heterogeneous multicore architectures.
*
* Copyright (C) 2015-2023 Université de Bordeaux, CNRS (LaBRI UMR 5800), Inria
*
* StarPU 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 2.1 of the License, or (at
* your option) any later version.
*
* StarPU 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 in COPYING.LGPL for more details.
*/
#include <starpu.h>
#include <omp.h>
#include <sched.h>
#if !defined(STARPU_PARALLEL_WORKER)
int main(void)
{
return 77;
}
#else
static void display_cpu(void *descr[], void *cl_arg)
{
(void)descr;
(void)cl_arg;
#pragma omp parallel
{
#ifdef __linux__
fprintf(stderr, "thread %d on cpu %d\n", omp_get_thread_num(), sched_getcpu());
#endif
}
}
static struct starpu_codelet display_cl =
{
.cpu_funcs = {display_cpu, NULL},
.nbuffers = 0,
};
void bind_func(void *arg)
{
(void) arg;
int workerid = starpu_worker_get_id_check();
if (starpu_worker_get_type(workerid) == STARPU_CPU_WORKER)
{
struct starpu_task *task = starpu_task_get_current();
int sched_ctx = task->sched_ctx;
int *cpuids = NULL;
int ncpuids = 0;
starpu_sched_ctx_get_available_cpuids(sched_ctx, &cpuids, &ncpuids);
omp_set_num_threads(ncpuids);
#pragma omp parallel
{
starpu_sched_ctx_bind_current_thread_to_cpuid(cpuids[omp_get_thread_num()]);
}
free(cpuids);
}
return;
}
int main(void)
{
int ret, i;
struct starpu_parallel_worker_config *parallel_workers;
setenv("STARPU_NMPI_MS","0",1);
ret = starpu_init(NULL);
if (ret == -ENODEV)
return 77;
STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
parallel_workers = starpu_parallel_worker_init(HWLOC_OBJ_SOCKET,
STARPU_PARALLEL_WORKER_POLICY_NAME, "dmdas",
STARPU_PARALLEL_WORKER_CREATE_FUNC, &bind_func,
STARPU_PARALLEL_WORKER_CREATE_FUNC_ARG, NULL,
0);
if (parallel_workers == NULL)
goto enodev;
starpu_parallel_worker_print(parallel_workers);
ret = starpu_task_insert(&display_cl, 0);
if (ret == -ENODEV)
goto enodev;
STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
/* wait for all tasks at the end*/
starpu_task_wait_for_all();
starpu_parallel_worker_shutdown(parallel_workers);
starpu_shutdown();
return 0;
enodev:
starpu_shutdown();
return 77;
}
#endif
|
