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
|
/* StarPU --- Runtime system for heterogeneous multicore architectures.
*
* Copyright (C) 2011, 2012 Centre National de la Recherche Scientifique
* Copyright (C) 2011 Université de Bordeaux 1
*
* 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_mpi_private.h>
#include <starpu_mpi_insert_task_cache.h>
#include <starpu_hash.h>
#include <common/htable32.h>
typedef struct _starpu_mpi_clear_cache_s {
starpu_data_handle_t data;
int rank;
int mode;
} _starpu_mpi_clear_cache_t;
struct starpu_htbl32_node **sent_data = NULL;
struct starpu_htbl32_node **received_data = NULL;
void _starpu_mpi_clear_cache_callback(void *callback_arg)
{
_starpu_mpi_clear_cache_t *clear_cache = (_starpu_mpi_clear_cache_t *)callback_arg;
uint32_t key = starpu_crc32_be((uintptr_t)clear_cache->data, 0);
if (clear_cache->mode == _STARPU_MPI_CLEAR_SENT_DATA) {
_STARPU_MPI_DEBUG("Clearing sent cache for data %p and rank %d\n", clear_cache->data, clear_cache->rank);
_starpu_htbl_insert_32(&sent_data[clear_cache->rank], key, NULL);
}
else if (clear_cache->mode == _STARPU_MPI_CLEAR_RECEIVED_DATA) {
_STARPU_MPI_DEBUG("Clearing received cache for data %p and rank %d\n", clear_cache->data, clear_cache->rank);
_starpu_htbl_insert_32(&received_data[clear_cache->rank], key, NULL);
}
free(clear_cache);
}
double _starpu_mpi_clear_cache_cost_function(struct starpu_task *task, unsigned nimpl)
{
return 0;
}
static struct starpu_perfmodel _starpu_mpi_clear_cache_model =
{
.cost_function = _starpu_mpi_clear_cache_cost_function,
.type = STARPU_COMMON,
};
static void _starpu_mpi_clear_cache_func(void *descr[] __attribute__ ((unused)), void *arg __attribute__ ((unused)))
{
}
static struct starpu_codelet _starpu_mpi_clear_cache_codelet =
{
.where = STARPU_CPU|STARPU_CUDA|STARPU_OPENCL,
.cpu_funcs = {_starpu_mpi_clear_cache_func, NULL},
.cuda_funcs = {_starpu_mpi_clear_cache_func, NULL},
.opencl_funcs = {_starpu_mpi_clear_cache_func, NULL},
.nbuffers = 1,
.modes = {STARPU_RW},
.model = &_starpu_mpi_clear_cache_model
// The model has a cost function which returns 0 so as to allow the codelet to be scheduled anywhere
};
void _starpu_mpi_clear_cache_request(starpu_data_handle_t data_handle, int rank, int mode)
{
struct starpu_task *task = starpu_task_create();
// We have a codelet with a empty function just to force the
// task being created to have a dependency on data_handle
task->cl = &_starpu_mpi_clear_cache_codelet;
task->handles[0] = data_handle;
_starpu_mpi_clear_cache_t *clear_cache = malloc(sizeof(_starpu_mpi_clear_cache_t));
clear_cache->data = data_handle;
clear_cache->rank = rank;
clear_cache->mode = mode;
task->callback_func = _starpu_mpi_clear_cache_callback;
task->callback_arg = clear_cache;
int ret = starpu_task_submit(task);
STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
}
|
