annotate libgomp/oacc-parallel.c @ 118:fd00160c1b76

ifdef TARGET_64BIT
author mir3636
date Tue, 27 Feb 2018 15:01:35 +0900
parents 04ced10e8804
children 84e7813d76e9
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
111
kono
parents:
diff changeset
1 /* Copyright (C) 2013-2017 Free Software Foundation, Inc.
kono
parents:
diff changeset
2
kono
parents:
diff changeset
3 Contributed by Mentor Embedded.
kono
parents:
diff changeset
4
kono
parents:
diff changeset
5 This file is part of the GNU Offloading and Multi Processing Library
kono
parents:
diff changeset
6 (libgomp).
kono
parents:
diff changeset
7
kono
parents:
diff changeset
8 Libgomp is free software; you can redistribute it and/or modify it
kono
parents:
diff changeset
9 under the terms of the GNU General Public License as published by
kono
parents:
diff changeset
10 the Free Software Foundation; either version 3, or (at your option)
kono
parents:
diff changeset
11 any later version.
kono
parents:
diff changeset
12
kono
parents:
diff changeset
13 Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY
kono
parents:
diff changeset
14 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
kono
parents:
diff changeset
15 FOR A PARTICULAR PURPOSE. See the GNU General Public License for
kono
parents:
diff changeset
16 more details.
kono
parents:
diff changeset
17
kono
parents:
diff changeset
18 Under Section 7 of GPL version 3, you are granted additional
kono
parents:
diff changeset
19 permissions described in the GCC Runtime Library Exception, version
kono
parents:
diff changeset
20 3.1, as published by the Free Software Foundation.
kono
parents:
diff changeset
21
kono
parents:
diff changeset
22 You should have received a copy of the GNU General Public License and
kono
parents:
diff changeset
23 a copy of the GCC Runtime Library Exception along with this program;
kono
parents:
diff changeset
24 see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
kono
parents:
diff changeset
25 <http://www.gnu.org/licenses/>. */
kono
parents:
diff changeset
26
kono
parents:
diff changeset
27 /* This file handles OpenACC constructs. */
kono
parents:
diff changeset
28
kono
parents:
diff changeset
29 #include "openacc.h"
kono
parents:
diff changeset
30 #include "libgomp.h"
kono
parents:
diff changeset
31 #include "libgomp_g.h"
kono
parents:
diff changeset
32 #include "gomp-constants.h"
kono
parents:
diff changeset
33 #include "oacc-int.h"
kono
parents:
diff changeset
34 #ifdef HAVE_INTTYPES_H
kono
parents:
diff changeset
35 # include <inttypes.h> /* For PRIu64. */
kono
parents:
diff changeset
36 #endif
kono
parents:
diff changeset
37 #include <string.h>
kono
parents:
diff changeset
38 #include <stdarg.h>
kono
parents:
diff changeset
39 #include <assert.h>
kono
parents:
diff changeset
40
kono
parents:
diff changeset
41 static int
kono
parents:
diff changeset
42 find_pset (int pos, size_t mapnum, unsigned short *kinds)
kono
parents:
diff changeset
43 {
kono
parents:
diff changeset
44 if (pos + 1 >= mapnum)
kono
parents:
diff changeset
45 return 0;
kono
parents:
diff changeset
46
kono
parents:
diff changeset
47 unsigned char kind = kinds[pos+1] & 0xff;
kono
parents:
diff changeset
48
kono
parents:
diff changeset
49 return kind == GOMP_MAP_TO_PSET;
kono
parents:
diff changeset
50 }
kono
parents:
diff changeset
51
kono
parents:
diff changeset
52 static void goacc_wait (int async, int num_waits, va_list *ap);
kono
parents:
diff changeset
53
kono
parents:
diff changeset
54
kono
parents:
diff changeset
55 /* Launch a possibly offloaded function on DEVICE. FN is the host fn
kono
parents:
diff changeset
56 address. MAPNUM, HOSTADDRS, SIZES & KINDS describe the memory
kono
parents:
diff changeset
57 blocks to be copied to/from the device. Varadic arguments are
kono
parents:
diff changeset
58 keyed optional parameters terminated with a zero. */
kono
parents:
diff changeset
59
kono
parents:
diff changeset
60 void
kono
parents:
diff changeset
61 GOACC_parallel_keyed (int device, void (*fn) (void *),
kono
parents:
diff changeset
62 size_t mapnum, void **hostaddrs, size_t *sizes,
kono
parents:
diff changeset
63 unsigned short *kinds, ...)
kono
parents:
diff changeset
64 {
kono
parents:
diff changeset
65 bool host_fallback = device == GOMP_DEVICE_HOST_FALLBACK;
kono
parents:
diff changeset
66 va_list ap;
kono
parents:
diff changeset
67 struct goacc_thread *thr;
kono
parents:
diff changeset
68 struct gomp_device_descr *acc_dev;
kono
parents:
diff changeset
69 struct target_mem_desc *tgt;
kono
parents:
diff changeset
70 void **devaddrs;
kono
parents:
diff changeset
71 unsigned int i;
kono
parents:
diff changeset
72 struct splay_tree_key_s k;
kono
parents:
diff changeset
73 splay_tree_key tgt_fn_key;
kono
parents:
diff changeset
74 void (*tgt_fn);
kono
parents:
diff changeset
75 int async = GOMP_ASYNC_SYNC;
kono
parents:
diff changeset
76 unsigned dims[GOMP_DIM_MAX];
kono
parents:
diff changeset
77 unsigned tag;
kono
parents:
diff changeset
78
kono
parents:
diff changeset
79 #ifdef HAVE_INTTYPES_H
kono
parents:
diff changeset
80 gomp_debug (0, "%s: mapnum=%"PRIu64", hostaddrs=%p, size=%p, kinds=%p\n",
kono
parents:
diff changeset
81 __FUNCTION__, (uint64_t) mapnum, hostaddrs, sizes, kinds);
kono
parents:
diff changeset
82 #else
kono
parents:
diff changeset
83 gomp_debug (0, "%s: mapnum=%lu, hostaddrs=%p, sizes=%p, kinds=%p\n",
kono
parents:
diff changeset
84 __FUNCTION__, (unsigned long) mapnum, hostaddrs, sizes, kinds);
kono
parents:
diff changeset
85 #endif
kono
parents:
diff changeset
86 goacc_lazy_initialize ();
kono
parents:
diff changeset
87
kono
parents:
diff changeset
88 thr = goacc_thread ();
kono
parents:
diff changeset
89 acc_dev = thr->dev;
kono
parents:
diff changeset
90
kono
parents:
diff changeset
91 /* Host fallback if "if" clause is false or if the current device is set to
kono
parents:
diff changeset
92 the host. */
kono
parents:
diff changeset
93 if (host_fallback)
kono
parents:
diff changeset
94 {
kono
parents:
diff changeset
95 goacc_save_and_set_bind (acc_device_host);
kono
parents:
diff changeset
96 fn (hostaddrs);
kono
parents:
diff changeset
97 goacc_restore_bind ();
kono
parents:
diff changeset
98 return;
kono
parents:
diff changeset
99 }
kono
parents:
diff changeset
100 else if (acc_device_type (acc_dev->type) == acc_device_host)
kono
parents:
diff changeset
101 {
kono
parents:
diff changeset
102 fn (hostaddrs);
kono
parents:
diff changeset
103 return;
kono
parents:
diff changeset
104 }
kono
parents:
diff changeset
105
kono
parents:
diff changeset
106 /* Default: let the runtime choose. */
kono
parents:
diff changeset
107 for (i = 0; i != GOMP_DIM_MAX; i++)
kono
parents:
diff changeset
108 dims[i] = 0;
kono
parents:
diff changeset
109
kono
parents:
diff changeset
110 va_start (ap, kinds);
kono
parents:
diff changeset
111 /* TODO: This will need amending when device_type is implemented. */
kono
parents:
diff changeset
112 while ((tag = va_arg (ap, unsigned)) != 0)
kono
parents:
diff changeset
113 {
kono
parents:
diff changeset
114 if (GOMP_LAUNCH_DEVICE (tag))
kono
parents:
diff changeset
115 gomp_fatal ("device_type '%d' offload parameters, libgomp is too old",
kono
parents:
diff changeset
116 GOMP_LAUNCH_DEVICE (tag));
kono
parents:
diff changeset
117
kono
parents:
diff changeset
118 switch (GOMP_LAUNCH_CODE (tag))
kono
parents:
diff changeset
119 {
kono
parents:
diff changeset
120 case GOMP_LAUNCH_DIM:
kono
parents:
diff changeset
121 {
kono
parents:
diff changeset
122 unsigned mask = GOMP_LAUNCH_OP (tag);
kono
parents:
diff changeset
123
kono
parents:
diff changeset
124 for (i = 0; i != GOMP_DIM_MAX; i++)
kono
parents:
diff changeset
125 if (mask & GOMP_DIM_MASK (i))
kono
parents:
diff changeset
126 dims[i] = va_arg (ap, unsigned);
kono
parents:
diff changeset
127 }
kono
parents:
diff changeset
128 break;
kono
parents:
diff changeset
129
kono
parents:
diff changeset
130 case GOMP_LAUNCH_ASYNC:
kono
parents:
diff changeset
131 {
kono
parents:
diff changeset
132 /* Small constant values are encoded in the operand. */
kono
parents:
diff changeset
133 async = GOMP_LAUNCH_OP (tag);
kono
parents:
diff changeset
134
kono
parents:
diff changeset
135 if (async == GOMP_LAUNCH_OP_MAX)
kono
parents:
diff changeset
136 async = va_arg (ap, unsigned);
kono
parents:
diff changeset
137 break;
kono
parents:
diff changeset
138 }
kono
parents:
diff changeset
139
kono
parents:
diff changeset
140 case GOMP_LAUNCH_WAIT:
kono
parents:
diff changeset
141 {
kono
parents:
diff changeset
142 unsigned num_waits = GOMP_LAUNCH_OP (tag);
kono
parents:
diff changeset
143
kono
parents:
diff changeset
144 if (num_waits)
kono
parents:
diff changeset
145 goacc_wait (async, num_waits, &ap);
kono
parents:
diff changeset
146 break;
kono
parents:
diff changeset
147 }
kono
parents:
diff changeset
148
kono
parents:
diff changeset
149 default:
kono
parents:
diff changeset
150 gomp_fatal ("unrecognized offload code '%d',"
kono
parents:
diff changeset
151 " libgomp is too old", GOMP_LAUNCH_CODE (tag));
kono
parents:
diff changeset
152 }
kono
parents:
diff changeset
153 }
kono
parents:
diff changeset
154 va_end (ap);
kono
parents:
diff changeset
155
kono
parents:
diff changeset
156 acc_dev->openacc.async_set_async_func (async);
kono
parents:
diff changeset
157
kono
parents:
diff changeset
158 if (!(acc_dev->capabilities & GOMP_OFFLOAD_CAP_NATIVE_EXEC))
kono
parents:
diff changeset
159 {
kono
parents:
diff changeset
160 k.host_start = (uintptr_t) fn;
kono
parents:
diff changeset
161 k.host_end = k.host_start + 1;
kono
parents:
diff changeset
162 gomp_mutex_lock (&acc_dev->lock);
kono
parents:
diff changeset
163 tgt_fn_key = splay_tree_lookup (&acc_dev->mem_map, &k);
kono
parents:
diff changeset
164 gomp_mutex_unlock (&acc_dev->lock);
kono
parents:
diff changeset
165
kono
parents:
diff changeset
166 if (tgt_fn_key == NULL)
kono
parents:
diff changeset
167 gomp_fatal ("target function wasn't mapped");
kono
parents:
diff changeset
168
kono
parents:
diff changeset
169 tgt_fn = (void (*)) tgt_fn_key->tgt_offset;
kono
parents:
diff changeset
170 }
kono
parents:
diff changeset
171 else
kono
parents:
diff changeset
172 tgt_fn = (void (*)) fn;
kono
parents:
diff changeset
173
kono
parents:
diff changeset
174 tgt = gomp_map_vars (acc_dev, mapnum, hostaddrs, NULL, sizes, kinds, true,
kono
parents:
diff changeset
175 GOMP_MAP_VARS_OPENACC);
kono
parents:
diff changeset
176
kono
parents:
diff changeset
177 devaddrs = gomp_alloca (sizeof (void *) * mapnum);
kono
parents:
diff changeset
178 for (i = 0; i < mapnum; i++)
kono
parents:
diff changeset
179 devaddrs[i] = (void *) (tgt->list[i].key->tgt->tgt_start
kono
parents:
diff changeset
180 + tgt->list[i].key->tgt_offset);
kono
parents:
diff changeset
181
kono
parents:
diff changeset
182 acc_dev->openacc.exec_func (tgt_fn, mapnum, hostaddrs, devaddrs,
kono
parents:
diff changeset
183 async, dims, tgt);
kono
parents:
diff changeset
184
kono
parents:
diff changeset
185 /* If running synchronously, unmap immediately. */
kono
parents:
diff changeset
186 if (async < acc_async_noval)
kono
parents:
diff changeset
187 gomp_unmap_vars (tgt, true);
kono
parents:
diff changeset
188 else
kono
parents:
diff changeset
189 tgt->device_descr->openacc.register_async_cleanup_func (tgt, async);
kono
parents:
diff changeset
190
kono
parents:
diff changeset
191 acc_dev->openacc.async_set_async_func (acc_async_sync);
kono
parents:
diff changeset
192 }
kono
parents:
diff changeset
193
kono
parents:
diff changeset
194 /* Legacy entry point, only provide host execution. */
kono
parents:
diff changeset
195
kono
parents:
diff changeset
196 void
kono
parents:
diff changeset
197 GOACC_parallel (int device, void (*fn) (void *),
kono
parents:
diff changeset
198 size_t mapnum, void **hostaddrs, size_t *sizes,
kono
parents:
diff changeset
199 unsigned short *kinds,
kono
parents:
diff changeset
200 int num_gangs, int num_workers, int vector_length,
kono
parents:
diff changeset
201 int async, int num_waits, ...)
kono
parents:
diff changeset
202 {
kono
parents:
diff changeset
203 goacc_save_and_set_bind (acc_device_host);
kono
parents:
diff changeset
204 fn (hostaddrs);
kono
parents:
diff changeset
205 goacc_restore_bind ();
kono
parents:
diff changeset
206 }
kono
parents:
diff changeset
207
kono
parents:
diff changeset
208 void
kono
parents:
diff changeset
209 GOACC_data_start (int device, size_t mapnum,
kono
parents:
diff changeset
210 void **hostaddrs, size_t *sizes, unsigned short *kinds)
kono
parents:
diff changeset
211 {
kono
parents:
diff changeset
212 bool host_fallback = device == GOMP_DEVICE_HOST_FALLBACK;
kono
parents:
diff changeset
213 struct target_mem_desc *tgt;
kono
parents:
diff changeset
214
kono
parents:
diff changeset
215 #ifdef HAVE_INTTYPES_H
kono
parents:
diff changeset
216 gomp_debug (0, "%s: mapnum=%"PRIu64", hostaddrs=%p, size=%p, kinds=%p\n",
kono
parents:
diff changeset
217 __FUNCTION__, (uint64_t) mapnum, hostaddrs, sizes, kinds);
kono
parents:
diff changeset
218 #else
kono
parents:
diff changeset
219 gomp_debug (0, "%s: mapnum=%lu, hostaddrs=%p, sizes=%p, kinds=%p\n",
kono
parents:
diff changeset
220 __FUNCTION__, (unsigned long) mapnum, hostaddrs, sizes, kinds);
kono
parents:
diff changeset
221 #endif
kono
parents:
diff changeset
222
kono
parents:
diff changeset
223 goacc_lazy_initialize ();
kono
parents:
diff changeset
224
kono
parents:
diff changeset
225 struct goacc_thread *thr = goacc_thread ();
kono
parents:
diff changeset
226 struct gomp_device_descr *acc_dev = thr->dev;
kono
parents:
diff changeset
227
kono
parents:
diff changeset
228 /* Host fallback or 'do nothing'. */
kono
parents:
diff changeset
229 if ((acc_dev->capabilities & GOMP_OFFLOAD_CAP_SHARED_MEM)
kono
parents:
diff changeset
230 || host_fallback)
kono
parents:
diff changeset
231 {
kono
parents:
diff changeset
232 tgt = gomp_map_vars (NULL, 0, NULL, NULL, NULL, NULL, true,
kono
parents:
diff changeset
233 GOMP_MAP_VARS_OPENACC);
kono
parents:
diff changeset
234 tgt->prev = thr->mapped_data;
kono
parents:
diff changeset
235 thr->mapped_data = tgt;
kono
parents:
diff changeset
236
kono
parents:
diff changeset
237 return;
kono
parents:
diff changeset
238 }
kono
parents:
diff changeset
239
kono
parents:
diff changeset
240 gomp_debug (0, " %s: prepare mappings\n", __FUNCTION__);
kono
parents:
diff changeset
241 tgt = gomp_map_vars (acc_dev, mapnum, hostaddrs, NULL, sizes, kinds, true,
kono
parents:
diff changeset
242 GOMP_MAP_VARS_OPENACC);
kono
parents:
diff changeset
243 gomp_debug (0, " %s: mappings prepared\n", __FUNCTION__);
kono
parents:
diff changeset
244 tgt->prev = thr->mapped_data;
kono
parents:
diff changeset
245 thr->mapped_data = tgt;
kono
parents:
diff changeset
246 }
kono
parents:
diff changeset
247
kono
parents:
diff changeset
248 void
kono
parents:
diff changeset
249 GOACC_data_end (void)
kono
parents:
diff changeset
250 {
kono
parents:
diff changeset
251 struct goacc_thread *thr = goacc_thread ();
kono
parents:
diff changeset
252 struct target_mem_desc *tgt = thr->mapped_data;
kono
parents:
diff changeset
253
kono
parents:
diff changeset
254 gomp_debug (0, " %s: restore mappings\n", __FUNCTION__);
kono
parents:
diff changeset
255 thr->mapped_data = tgt->prev;
kono
parents:
diff changeset
256 gomp_unmap_vars (tgt, true);
kono
parents:
diff changeset
257 gomp_debug (0, " %s: mappings restored\n", __FUNCTION__);
kono
parents:
diff changeset
258 }
kono
parents:
diff changeset
259
kono
parents:
diff changeset
260 void
kono
parents:
diff changeset
261 GOACC_enter_exit_data (int device, size_t mapnum,
kono
parents:
diff changeset
262 void **hostaddrs, size_t *sizes, unsigned short *kinds,
kono
parents:
diff changeset
263 int async, int num_waits, ...)
kono
parents:
diff changeset
264 {
kono
parents:
diff changeset
265 struct goacc_thread *thr;
kono
parents:
diff changeset
266 struct gomp_device_descr *acc_dev;
kono
parents:
diff changeset
267 bool host_fallback = device == GOMP_DEVICE_HOST_FALLBACK;
kono
parents:
diff changeset
268 bool data_enter = false;
kono
parents:
diff changeset
269 size_t i;
kono
parents:
diff changeset
270
kono
parents:
diff changeset
271 goacc_lazy_initialize ();
kono
parents:
diff changeset
272
kono
parents:
diff changeset
273 thr = goacc_thread ();
kono
parents:
diff changeset
274 acc_dev = thr->dev;
kono
parents:
diff changeset
275
kono
parents:
diff changeset
276 if ((acc_dev->capabilities & GOMP_OFFLOAD_CAP_SHARED_MEM)
kono
parents:
diff changeset
277 || host_fallback)
kono
parents:
diff changeset
278 return;
kono
parents:
diff changeset
279
kono
parents:
diff changeset
280 if (num_waits)
kono
parents:
diff changeset
281 {
kono
parents:
diff changeset
282 va_list ap;
kono
parents:
diff changeset
283
kono
parents:
diff changeset
284 va_start (ap, num_waits);
kono
parents:
diff changeset
285 goacc_wait (async, num_waits, &ap);
kono
parents:
diff changeset
286 va_end (ap);
kono
parents:
diff changeset
287 }
kono
parents:
diff changeset
288
kono
parents:
diff changeset
289 acc_dev->openacc.async_set_async_func (async);
kono
parents:
diff changeset
290
kono
parents:
diff changeset
291 /* Determine if this is an "acc enter data". */
kono
parents:
diff changeset
292 for (i = 0; i < mapnum; ++i)
kono
parents:
diff changeset
293 {
kono
parents:
diff changeset
294 unsigned char kind = kinds[i] & 0xff;
kono
parents:
diff changeset
295
kono
parents:
diff changeset
296 if (kind == GOMP_MAP_POINTER || kind == GOMP_MAP_TO_PSET)
kono
parents:
diff changeset
297 continue;
kono
parents:
diff changeset
298
kono
parents:
diff changeset
299 if (kind == GOMP_MAP_FORCE_ALLOC
kono
parents:
diff changeset
300 || kind == GOMP_MAP_FORCE_PRESENT
kono
parents:
diff changeset
301 || kind == GOMP_MAP_FORCE_TO)
kono
parents:
diff changeset
302 {
kono
parents:
diff changeset
303 data_enter = true;
kono
parents:
diff changeset
304 break;
kono
parents:
diff changeset
305 }
kono
parents:
diff changeset
306
kono
parents:
diff changeset
307 if (kind == GOMP_MAP_DELETE
kono
parents:
diff changeset
308 || kind == GOMP_MAP_FORCE_FROM)
kono
parents:
diff changeset
309 break;
kono
parents:
diff changeset
310
kono
parents:
diff changeset
311 gomp_fatal (">>>> GOACC_enter_exit_data UNHANDLED kind 0x%.2x",
kono
parents:
diff changeset
312 kind);
kono
parents:
diff changeset
313 }
kono
parents:
diff changeset
314
kono
parents:
diff changeset
315 if (data_enter)
kono
parents:
diff changeset
316 {
kono
parents:
diff changeset
317 for (i = 0; i < mapnum; i++)
kono
parents:
diff changeset
318 {
kono
parents:
diff changeset
319 unsigned char kind = kinds[i] & 0xff;
kono
parents:
diff changeset
320
kono
parents:
diff changeset
321 /* Scan for PSETs. */
kono
parents:
diff changeset
322 int psets = find_pset (i, mapnum, kinds);
kono
parents:
diff changeset
323
kono
parents:
diff changeset
324 if (!psets)
kono
parents:
diff changeset
325 {
kono
parents:
diff changeset
326 switch (kind)
kono
parents:
diff changeset
327 {
kono
parents:
diff changeset
328 case GOMP_MAP_POINTER:
kono
parents:
diff changeset
329 gomp_acc_insert_pointer (1, &hostaddrs[i], &sizes[i],
kono
parents:
diff changeset
330 &kinds[i]);
kono
parents:
diff changeset
331 break;
kono
parents:
diff changeset
332 case GOMP_MAP_FORCE_ALLOC:
kono
parents:
diff changeset
333 acc_create (hostaddrs[i], sizes[i]);
kono
parents:
diff changeset
334 break;
kono
parents:
diff changeset
335 case GOMP_MAP_FORCE_PRESENT:
kono
parents:
diff changeset
336 acc_present_or_copyin (hostaddrs[i], sizes[i]);
kono
parents:
diff changeset
337 break;
kono
parents:
diff changeset
338 case GOMP_MAP_FORCE_TO:
kono
parents:
diff changeset
339 acc_present_or_copyin (hostaddrs[i], sizes[i]);
kono
parents:
diff changeset
340 break;
kono
parents:
diff changeset
341 default:
kono
parents:
diff changeset
342 gomp_fatal (">>>> GOACC_enter_exit_data UNHANDLED kind 0x%.2x",
kono
parents:
diff changeset
343 kind);
kono
parents:
diff changeset
344 break;
kono
parents:
diff changeset
345 }
kono
parents:
diff changeset
346 }
kono
parents:
diff changeset
347 else
kono
parents:
diff changeset
348 {
kono
parents:
diff changeset
349 gomp_acc_insert_pointer (3, &hostaddrs[i], &sizes[i], &kinds[i]);
kono
parents:
diff changeset
350 /* Increment 'i' by two because OpenACC requires fortran
kono
parents:
diff changeset
351 arrays to be contiguous, so each PSET is associated with
kono
parents:
diff changeset
352 one of MAP_FORCE_ALLOC/MAP_FORCE_PRESET/MAP_FORCE_TO, and
kono
parents:
diff changeset
353 one MAP_POINTER. */
kono
parents:
diff changeset
354 i += 2;
kono
parents:
diff changeset
355 }
kono
parents:
diff changeset
356 }
kono
parents:
diff changeset
357 }
kono
parents:
diff changeset
358 else
kono
parents:
diff changeset
359 for (i = 0; i < mapnum; ++i)
kono
parents:
diff changeset
360 {
kono
parents:
diff changeset
361 unsigned char kind = kinds[i] & 0xff;
kono
parents:
diff changeset
362
kono
parents:
diff changeset
363 int psets = find_pset (i, mapnum, kinds);
kono
parents:
diff changeset
364
kono
parents:
diff changeset
365 if (!psets)
kono
parents:
diff changeset
366 {
kono
parents:
diff changeset
367 switch (kind)
kono
parents:
diff changeset
368 {
kono
parents:
diff changeset
369 case GOMP_MAP_POINTER:
kono
parents:
diff changeset
370 gomp_acc_remove_pointer (hostaddrs[i], (kinds[i] & 0xff)
kono
parents:
diff changeset
371 == GOMP_MAP_FORCE_FROM,
kono
parents:
diff changeset
372 async, 1);
kono
parents:
diff changeset
373 break;
kono
parents:
diff changeset
374 case GOMP_MAP_DELETE:
kono
parents:
diff changeset
375 acc_delete (hostaddrs[i], sizes[i]);
kono
parents:
diff changeset
376 break;
kono
parents:
diff changeset
377 case GOMP_MAP_FORCE_FROM:
kono
parents:
diff changeset
378 acc_copyout (hostaddrs[i], sizes[i]);
kono
parents:
diff changeset
379 break;
kono
parents:
diff changeset
380 default:
kono
parents:
diff changeset
381 gomp_fatal (">>>> GOACC_enter_exit_data UNHANDLED kind 0x%.2x",
kono
parents:
diff changeset
382 kind);
kono
parents:
diff changeset
383 break;
kono
parents:
diff changeset
384 }
kono
parents:
diff changeset
385 }
kono
parents:
diff changeset
386 else
kono
parents:
diff changeset
387 {
kono
parents:
diff changeset
388 gomp_acc_remove_pointer (hostaddrs[i], (kinds[i] & 0xff)
kono
parents:
diff changeset
389 == GOMP_MAP_FORCE_FROM, async, 3);
kono
parents:
diff changeset
390 /* See the above comment. */
kono
parents:
diff changeset
391 i += 2;
kono
parents:
diff changeset
392 }
kono
parents:
diff changeset
393 }
kono
parents:
diff changeset
394
kono
parents:
diff changeset
395 acc_dev->openacc.async_set_async_func (acc_async_sync);
kono
parents:
diff changeset
396 }
kono
parents:
diff changeset
397
kono
parents:
diff changeset
398 static void
kono
parents:
diff changeset
399 goacc_wait (int async, int num_waits, va_list *ap)
kono
parents:
diff changeset
400 {
kono
parents:
diff changeset
401 struct goacc_thread *thr = goacc_thread ();
kono
parents:
diff changeset
402 struct gomp_device_descr *acc_dev = thr->dev;
kono
parents:
diff changeset
403
kono
parents:
diff changeset
404 while (num_waits--)
kono
parents:
diff changeset
405 {
kono
parents:
diff changeset
406 int qid = va_arg (*ap, int);
kono
parents:
diff changeset
407
kono
parents:
diff changeset
408 if (acc_async_test (qid))
kono
parents:
diff changeset
409 continue;
kono
parents:
diff changeset
410
kono
parents:
diff changeset
411 if (async == acc_async_sync)
kono
parents:
diff changeset
412 acc_wait (qid);
kono
parents:
diff changeset
413 else if (qid == async)
kono
parents:
diff changeset
414 ;/* If we're waiting on the same asynchronous queue as we're
kono
parents:
diff changeset
415 launching on, the queue itself will order work as
kono
parents:
diff changeset
416 required, so there's no need to wait explicitly. */
kono
parents:
diff changeset
417 else
kono
parents:
diff changeset
418 acc_dev->openacc.async_wait_async_func (qid, async);
kono
parents:
diff changeset
419 }
kono
parents:
diff changeset
420 }
kono
parents:
diff changeset
421
kono
parents:
diff changeset
422 void
kono
parents:
diff changeset
423 GOACC_update (int device, size_t mapnum,
kono
parents:
diff changeset
424 void **hostaddrs, size_t *sizes, unsigned short *kinds,
kono
parents:
diff changeset
425 int async, int num_waits, ...)
kono
parents:
diff changeset
426 {
kono
parents:
diff changeset
427 bool host_fallback = device == GOMP_DEVICE_HOST_FALLBACK;
kono
parents:
diff changeset
428 size_t i;
kono
parents:
diff changeset
429
kono
parents:
diff changeset
430 goacc_lazy_initialize ();
kono
parents:
diff changeset
431
kono
parents:
diff changeset
432 struct goacc_thread *thr = goacc_thread ();
kono
parents:
diff changeset
433 struct gomp_device_descr *acc_dev = thr->dev;
kono
parents:
diff changeset
434
kono
parents:
diff changeset
435 if ((acc_dev->capabilities & GOMP_OFFLOAD_CAP_SHARED_MEM)
kono
parents:
diff changeset
436 || host_fallback)
kono
parents:
diff changeset
437 return;
kono
parents:
diff changeset
438
kono
parents:
diff changeset
439 if (num_waits)
kono
parents:
diff changeset
440 {
kono
parents:
diff changeset
441 va_list ap;
kono
parents:
diff changeset
442
kono
parents:
diff changeset
443 va_start (ap, num_waits);
kono
parents:
diff changeset
444 goacc_wait (async, num_waits, &ap);
kono
parents:
diff changeset
445 va_end (ap);
kono
parents:
diff changeset
446 }
kono
parents:
diff changeset
447
kono
parents:
diff changeset
448 acc_dev->openacc.async_set_async_func (async);
kono
parents:
diff changeset
449
kono
parents:
diff changeset
450 for (i = 0; i < mapnum; ++i)
kono
parents:
diff changeset
451 {
kono
parents:
diff changeset
452 unsigned char kind = kinds[i] & 0xff;
kono
parents:
diff changeset
453
kono
parents:
diff changeset
454 switch (kind)
kono
parents:
diff changeset
455 {
kono
parents:
diff changeset
456 case GOMP_MAP_POINTER:
kono
parents:
diff changeset
457 case GOMP_MAP_TO_PSET:
kono
parents:
diff changeset
458 break;
kono
parents:
diff changeset
459
kono
parents:
diff changeset
460 case GOMP_MAP_FORCE_TO:
kono
parents:
diff changeset
461 acc_update_device (hostaddrs[i], sizes[i]);
kono
parents:
diff changeset
462 break;
kono
parents:
diff changeset
463
kono
parents:
diff changeset
464 case GOMP_MAP_FORCE_FROM:
kono
parents:
diff changeset
465 acc_update_self (hostaddrs[i], sizes[i]);
kono
parents:
diff changeset
466 break;
kono
parents:
diff changeset
467
kono
parents:
diff changeset
468 default:
kono
parents:
diff changeset
469 gomp_fatal (">>>> GOACC_update UNHANDLED kind 0x%.2x", kind);
kono
parents:
diff changeset
470 break;
kono
parents:
diff changeset
471 }
kono
parents:
diff changeset
472 }
kono
parents:
diff changeset
473
kono
parents:
diff changeset
474 acc_dev->openacc.async_set_async_func (acc_async_sync);
kono
parents:
diff changeset
475 }
kono
parents:
diff changeset
476
kono
parents:
diff changeset
477 void
kono
parents:
diff changeset
478 GOACC_wait (int async, int num_waits, ...)
kono
parents:
diff changeset
479 {
kono
parents:
diff changeset
480 if (num_waits)
kono
parents:
diff changeset
481 {
kono
parents:
diff changeset
482 va_list ap;
kono
parents:
diff changeset
483
kono
parents:
diff changeset
484 va_start (ap, num_waits);
kono
parents:
diff changeset
485 goacc_wait (async, num_waits, &ap);
kono
parents:
diff changeset
486 va_end (ap);
kono
parents:
diff changeset
487 }
kono
parents:
diff changeset
488 else if (async == acc_async_sync)
kono
parents:
diff changeset
489 acc_wait_all ();
kono
parents:
diff changeset
490 else if (async == acc_async_noval)
kono
parents:
diff changeset
491 goacc_thread ()->dev->openacc.async_wait_all_async_func (acc_async_noval);
kono
parents:
diff changeset
492 }
kono
parents:
diff changeset
493
kono
parents:
diff changeset
494 int
kono
parents:
diff changeset
495 GOACC_get_num_threads (void)
kono
parents:
diff changeset
496 {
kono
parents:
diff changeset
497 return 1;
kono
parents:
diff changeset
498 }
kono
parents:
diff changeset
499
kono
parents:
diff changeset
500 int
kono
parents:
diff changeset
501 GOACC_get_thread_num (void)
kono
parents:
diff changeset
502 {
kono
parents:
diff changeset
503 return 0;
kono
parents:
diff changeset
504 }
kono
parents:
diff changeset
505
kono
parents:
diff changeset
506 void
kono
parents:
diff changeset
507 GOACC_declare (int device, size_t mapnum,
kono
parents:
diff changeset
508 void **hostaddrs, size_t *sizes, unsigned short *kinds)
kono
parents:
diff changeset
509 {
kono
parents:
diff changeset
510 int i;
kono
parents:
diff changeset
511
kono
parents:
diff changeset
512 for (i = 0; i < mapnum; i++)
kono
parents:
diff changeset
513 {
kono
parents:
diff changeset
514 unsigned char kind = kinds[i] & 0xff;
kono
parents:
diff changeset
515
kono
parents:
diff changeset
516 if (kind == GOMP_MAP_POINTER || kind == GOMP_MAP_TO_PSET)
kono
parents:
diff changeset
517 continue;
kono
parents:
diff changeset
518
kono
parents:
diff changeset
519 switch (kind)
kono
parents:
diff changeset
520 {
kono
parents:
diff changeset
521 case GOMP_MAP_FORCE_ALLOC:
kono
parents:
diff changeset
522 case GOMP_MAP_FORCE_FROM:
kono
parents:
diff changeset
523 case GOMP_MAP_FORCE_TO:
kono
parents:
diff changeset
524 case GOMP_MAP_POINTER:
kono
parents:
diff changeset
525 case GOMP_MAP_DELETE:
kono
parents:
diff changeset
526 GOACC_enter_exit_data (device, 1, &hostaddrs[i], &sizes[i],
kono
parents:
diff changeset
527 &kinds[i], 0, 0);
kono
parents:
diff changeset
528 break;
kono
parents:
diff changeset
529
kono
parents:
diff changeset
530 case GOMP_MAP_FORCE_DEVICEPTR:
kono
parents:
diff changeset
531 break;
kono
parents:
diff changeset
532
kono
parents:
diff changeset
533 case GOMP_MAP_ALLOC:
kono
parents:
diff changeset
534 if (!acc_is_present (hostaddrs[i], sizes[i]))
kono
parents:
diff changeset
535 GOACC_enter_exit_data (device, 1, &hostaddrs[i], &sizes[i],
kono
parents:
diff changeset
536 &kinds[i], 0, 0);
kono
parents:
diff changeset
537 break;
kono
parents:
diff changeset
538
kono
parents:
diff changeset
539 case GOMP_MAP_TO:
kono
parents:
diff changeset
540 GOACC_enter_exit_data (device, 1, &hostaddrs[i], &sizes[i],
kono
parents:
diff changeset
541 &kinds[i], 0, 0);
kono
parents:
diff changeset
542
kono
parents:
diff changeset
543 break;
kono
parents:
diff changeset
544
kono
parents:
diff changeset
545 case GOMP_MAP_FROM:
kono
parents:
diff changeset
546 kinds[i] = GOMP_MAP_FORCE_FROM;
kono
parents:
diff changeset
547 GOACC_enter_exit_data (device, 1, &hostaddrs[i], &sizes[i],
kono
parents:
diff changeset
548 &kinds[i], 0, 0);
kono
parents:
diff changeset
549 break;
kono
parents:
diff changeset
550
kono
parents:
diff changeset
551 case GOMP_MAP_FORCE_PRESENT:
kono
parents:
diff changeset
552 if (!acc_is_present (hostaddrs[i], sizes[i]))
kono
parents:
diff changeset
553 gomp_fatal ("[%p,%ld] is not mapped", hostaddrs[i],
kono
parents:
diff changeset
554 (unsigned long) sizes[i]);
kono
parents:
diff changeset
555 break;
kono
parents:
diff changeset
556
kono
parents:
diff changeset
557 default:
kono
parents:
diff changeset
558 assert (0);
kono
parents:
diff changeset
559 break;
kono
parents:
diff changeset
560 }
kono
parents:
diff changeset
561 }
kono
parents:
diff changeset
562 }