1 /* Copyright (C) 2013-2018 Free Software Foundation, Inc.
3 Contributed by Mentor Embedded.
5 This file is part of the GNU Offloading and Multi Processing Library
8 Libgomp is free software; you can redistribute it and/or modify it
9 under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3, or (at your option)
13 Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY
14 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
15 FOR A PARTICULAR PURPOSE. See the GNU General Public License for
18 Under Section 7 of GPL version 3, you are granted additional
19 permissions described in the GCC Runtime Library Exception, version
20 3.1, as published by the Free Software Foundation.
22 You should have received a copy of the GNU General Public License and
23 a copy of the GCC Runtime Library Exception along with this program;
24 see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
25 <http://www.gnu.org/licenses/>. */
27 /* This file handles OpenACC constructs. */
31 #include "libgomp_g.h"
32 #include "gomp-constants.h"
34 #ifdef HAVE_INTTYPES_H
35 # include <inttypes.h> /* For PRIu64. */
41 /* Returns the number of mappings associated with the pointer or pset. PSET
42 have three mappings, whereas pointer have two. */
45 find_pointer (int pos
, size_t mapnum
, unsigned short *kinds
)
47 if (pos
+ 1 >= mapnum
)
50 unsigned char kind
= kinds
[pos
+1] & 0xff;
52 if (kind
== GOMP_MAP_TO_PSET
)
54 else if (kind
== GOMP_MAP_POINTER
)
60 /* Handle the mapping pair that are presented when a
61 deviceptr clause is used with Fortran. */
64 handle_ftn_pointers (size_t mapnum
, void **hostaddrs
, size_t *sizes
,
65 unsigned short *kinds
)
69 for (i
= 0; i
< mapnum
; i
++)
71 unsigned short kind1
= kinds
[i
] & 0xff;
73 /* Handle Fortran deviceptr clause. */
74 if (kind1
== GOMP_MAP_FORCE_DEVICEPTR
)
78 if (i
< (signed)mapnum
- 1)
79 kind2
= kinds
[i
+ 1] & 0xff;
83 if (sizes
[i
] == sizeof (void *))
86 /* At this point, we're dealing with a Fortran deviceptr.
87 If the next element is not what we're expecting, then
88 this is an instance of where the deviceptr variable was
89 not used within the region and the pointer was removed
91 if (kind2
== GOMP_MAP_POINTER
93 && hostaddrs
[i
] == *(void **)hostaddrs
[i
+ 1])
95 kinds
[i
+1] = kinds
[i
];
96 sizes
[i
+1] = sizeof (void *);
99 /* Invalidate the entry. */
105 static void goacc_wait (int async
, int num_waits
, va_list *ap
);
108 /* Launch a possibly offloaded function on DEVICE. FN is the host fn
109 address. MAPNUM, HOSTADDRS, SIZES & KINDS describe the memory
110 blocks to be copied to/from the device. Varadic arguments are
111 keyed optional parameters terminated with a zero. */
114 GOACC_parallel_keyed (int device
, void (*fn
) (void *),
115 size_t mapnum
, void **hostaddrs
, size_t *sizes
,
116 unsigned short *kinds
, ...)
118 bool host_fallback
= device
== GOMP_DEVICE_HOST_FALLBACK
;
120 struct goacc_thread
*thr
;
121 struct gomp_device_descr
*acc_dev
;
122 struct target_mem_desc
*tgt
;
125 struct splay_tree_key_s k
;
126 splay_tree_key tgt_fn_key
;
128 int async
= GOMP_ASYNC_SYNC
;
129 unsigned dims
[GOMP_DIM_MAX
];
132 #ifdef HAVE_INTTYPES_H
133 gomp_debug (0, "%s: mapnum=%"PRIu64
", hostaddrs=%p, size=%p, kinds=%p\n",
134 __FUNCTION__
, (uint64_t) mapnum
, hostaddrs
, sizes
, kinds
);
136 gomp_debug (0, "%s: mapnum=%lu, hostaddrs=%p, sizes=%p, kinds=%p\n",
137 __FUNCTION__
, (unsigned long) mapnum
, hostaddrs
, sizes
, kinds
);
139 goacc_lazy_initialize ();
141 thr
= goacc_thread ();
144 handle_ftn_pointers (mapnum
, hostaddrs
, sizes
, kinds
);
146 /* Host fallback if "if" clause is false or if the current device is set to
150 goacc_save_and_set_bind (acc_device_host
);
152 goacc_restore_bind ();
155 else if (acc_device_type (acc_dev
->type
) == acc_device_host
)
161 /* Default: let the runtime choose. */
162 for (i
= 0; i
!= GOMP_DIM_MAX
; i
++)
165 va_start (ap
, kinds
);
166 /* TODO: This will need amending when device_type is implemented. */
167 while ((tag
= va_arg (ap
, unsigned)) != 0)
169 if (GOMP_LAUNCH_DEVICE (tag
))
170 gomp_fatal ("device_type '%d' offload parameters, libgomp is too old",
171 GOMP_LAUNCH_DEVICE (tag
));
173 switch (GOMP_LAUNCH_CODE (tag
))
175 case GOMP_LAUNCH_DIM
:
177 unsigned mask
= GOMP_LAUNCH_OP (tag
);
179 for (i
= 0; i
!= GOMP_DIM_MAX
; i
++)
180 if (mask
& GOMP_DIM_MASK (i
))
181 dims
[i
] = va_arg (ap
, unsigned);
185 case GOMP_LAUNCH_ASYNC
:
187 /* Small constant values are encoded in the operand. */
188 async
= GOMP_LAUNCH_OP (tag
);
190 if (async
== GOMP_LAUNCH_OP_MAX
)
191 async
= va_arg (ap
, unsigned);
195 case GOMP_LAUNCH_WAIT
:
197 unsigned num_waits
= GOMP_LAUNCH_OP (tag
);
200 goacc_wait (async
, num_waits
, &ap
);
205 gomp_fatal ("unrecognized offload code '%d',"
206 " libgomp is too old", GOMP_LAUNCH_CODE (tag
));
211 acc_dev
->openacc
.async_set_async_func (async
);
213 if (!(acc_dev
->capabilities
& GOMP_OFFLOAD_CAP_NATIVE_EXEC
))
215 k
.host_start
= (uintptr_t) fn
;
216 k
.host_end
= k
.host_start
+ 1;
217 gomp_mutex_lock (&acc_dev
->lock
);
218 tgt_fn_key
= splay_tree_lookup (&acc_dev
->mem_map
, &k
);
219 gomp_mutex_unlock (&acc_dev
->lock
);
221 if (tgt_fn_key
== NULL
)
222 gomp_fatal ("target function wasn't mapped");
224 tgt_fn
= (void (*)) tgt_fn_key
->tgt_offset
;
227 tgt_fn
= (void (*)) fn
;
229 tgt
= gomp_map_vars (acc_dev
, mapnum
, hostaddrs
, NULL
, sizes
, kinds
, true,
230 GOMP_MAP_VARS_OPENACC
);
232 devaddrs
= gomp_alloca (sizeof (void *) * mapnum
);
233 for (i
= 0; i
< mapnum
; i
++)
234 devaddrs
[i
] = (void *) (tgt
->list
[i
].key
->tgt
->tgt_start
235 + tgt
->list
[i
].key
->tgt_offset
);
237 acc_dev
->openacc
.exec_func (tgt_fn
, mapnum
, hostaddrs
, devaddrs
,
240 /* If running synchronously, unmap immediately. */
241 bool copyfrom
= true;
242 if (async_synchronous_p (async
))
243 gomp_unmap_vars (tgt
, true);
246 bool async_unmap
= false;
247 for (size_t i
= 0; i
< tgt
->list_count
; i
++)
249 splay_tree_key k
= tgt
->list
[i
].key
;
250 if (k
&& k
->refcount
== 1)
257 tgt
->device_descr
->openacc
.register_async_cleanup_func (tgt
, async
);
261 gomp_unmap_vars (tgt
, copyfrom
);
265 acc_dev
->openacc
.async_set_async_func (acc_async_sync
);
268 /* Legacy entry point, only provide host execution. */
271 GOACC_parallel (int device
, void (*fn
) (void *),
272 size_t mapnum
, void **hostaddrs
, size_t *sizes
,
273 unsigned short *kinds
,
274 int num_gangs
, int num_workers
, int vector_length
,
275 int async
, int num_waits
, ...)
277 goacc_save_and_set_bind (acc_device_host
);
279 goacc_restore_bind ();
283 GOACC_data_start (int device
, size_t mapnum
,
284 void **hostaddrs
, size_t *sizes
, unsigned short *kinds
)
286 bool host_fallback
= device
== GOMP_DEVICE_HOST_FALLBACK
;
287 struct target_mem_desc
*tgt
;
289 #ifdef HAVE_INTTYPES_H
290 gomp_debug (0, "%s: mapnum=%"PRIu64
", hostaddrs=%p, size=%p, kinds=%p\n",
291 __FUNCTION__
, (uint64_t) mapnum
, hostaddrs
, sizes
, kinds
);
293 gomp_debug (0, "%s: mapnum=%lu, hostaddrs=%p, sizes=%p, kinds=%p\n",
294 __FUNCTION__
, (unsigned long) mapnum
, hostaddrs
, sizes
, kinds
);
297 goacc_lazy_initialize ();
299 struct goacc_thread
*thr
= goacc_thread ();
300 struct gomp_device_descr
*acc_dev
= thr
->dev
;
302 /* Host fallback or 'do nothing'. */
303 if ((acc_dev
->capabilities
& GOMP_OFFLOAD_CAP_SHARED_MEM
)
306 tgt
= gomp_map_vars (NULL
, 0, NULL
, NULL
, NULL
, NULL
, true,
307 GOMP_MAP_VARS_OPENACC
);
308 tgt
->prev
= thr
->mapped_data
;
309 thr
->mapped_data
= tgt
;
314 gomp_debug (0, " %s: prepare mappings\n", __FUNCTION__
);
315 tgt
= gomp_map_vars (acc_dev
, mapnum
, hostaddrs
, NULL
, sizes
, kinds
, true,
316 GOMP_MAP_VARS_OPENACC
);
317 gomp_debug (0, " %s: mappings prepared\n", __FUNCTION__
);
318 tgt
->prev
= thr
->mapped_data
;
319 thr
->mapped_data
= tgt
;
323 GOACC_data_end (void)
325 struct goacc_thread
*thr
= goacc_thread ();
326 struct target_mem_desc
*tgt
= thr
->mapped_data
;
328 gomp_debug (0, " %s: restore mappings\n", __FUNCTION__
);
329 thr
->mapped_data
= tgt
->prev
;
330 gomp_unmap_vars (tgt
, true);
331 gomp_debug (0, " %s: mappings restored\n", __FUNCTION__
);
335 GOACC_enter_exit_data (int device
, size_t mapnum
,
336 void **hostaddrs
, size_t *sizes
, unsigned short *kinds
,
337 int async
, int num_waits
, ...)
339 struct goacc_thread
*thr
;
340 struct gomp_device_descr
*acc_dev
;
341 bool host_fallback
= device
== GOMP_DEVICE_HOST_FALLBACK
;
342 bool data_enter
= false;
345 goacc_lazy_initialize ();
347 thr
= goacc_thread ();
350 if ((acc_dev
->capabilities
& GOMP_OFFLOAD_CAP_SHARED_MEM
)
358 va_start (ap
, num_waits
);
359 goacc_wait (async
, num_waits
, &ap
);
363 /* Determine whether "finalize" semantics apply to all mappings of this
364 OpenACC directive. */
365 bool finalize
= false;
368 unsigned char kind
= kinds
[0] & 0xff;
369 if (kind
== GOMP_MAP_DELETE
370 || kind
== GOMP_MAP_FORCE_FROM
)
374 acc_dev
->openacc
.async_set_async_func (async
);
376 /* Determine if this is an "acc enter data". */
377 for (i
= 0; i
< mapnum
; ++i
)
379 unsigned char kind
= kinds
[i
] & 0xff;
381 if (kind
== GOMP_MAP_POINTER
|| kind
== GOMP_MAP_TO_PSET
)
384 if (kind
== GOMP_MAP_FORCE_ALLOC
385 || kind
== GOMP_MAP_FORCE_PRESENT
386 || kind
== GOMP_MAP_FORCE_TO
387 || kind
== GOMP_MAP_TO
388 || kind
== GOMP_MAP_ALLOC
)
394 if (kind
== GOMP_MAP_RELEASE
395 || kind
== GOMP_MAP_DELETE
396 || kind
== GOMP_MAP_FROM
397 || kind
== GOMP_MAP_FORCE_FROM
)
400 gomp_fatal (">>>> GOACC_enter_exit_data UNHANDLED kind 0x%.2x",
404 /* In c, non-pointers and arrays are represented by a single data clause.
405 Dynamically allocated arrays and subarrays are represented by a data
406 clause followed by an internal GOMP_MAP_POINTER.
408 In fortran, scalars and not allocated arrays are represented by a
409 single data clause. Allocated arrays and subarrays have three mappings:
410 1) the original data clause, 2) a PSET 3) a pointer to the array data.
415 for (i
= 0; i
< mapnum
; i
++)
417 unsigned char kind
= kinds
[i
] & 0xff;
419 /* Scan for pointers and PSETs. */
420 int pointer
= find_pointer (i
, mapnum
, kinds
);
427 acc_present_or_create (hostaddrs
[i
], sizes
[i
]);
429 case GOMP_MAP_FORCE_ALLOC
:
430 acc_create (hostaddrs
[i
], sizes
[i
]);
433 acc_present_or_copyin (hostaddrs
[i
], sizes
[i
]);
435 case GOMP_MAP_FORCE_TO
:
436 acc_copyin (hostaddrs
[i
], sizes
[i
]);
439 gomp_fatal (">>>> GOACC_enter_exit_data UNHANDLED kind 0x%.2x",
446 gomp_acc_insert_pointer (pointer
, &hostaddrs
[i
],
447 &sizes
[i
], &kinds
[i
]);
448 /* Increment 'i' by two because OpenACC requires fortran
449 arrays to be contiguous, so each PSET is associated with
450 one of MAP_FORCE_ALLOC/MAP_FORCE_PRESET/MAP_FORCE_TO, and
457 for (i
= 0; i
< mapnum
; ++i
)
459 unsigned char kind
= kinds
[i
] & 0xff;
461 int pointer
= find_pointer (i
, mapnum
, kinds
);
467 case GOMP_MAP_RELEASE
:
468 case GOMP_MAP_DELETE
:
469 if (acc_is_present (hostaddrs
[i
], sizes
[i
]))
472 acc_delete_finalize (hostaddrs
[i
], sizes
[i
]);
474 acc_delete (hostaddrs
[i
], sizes
[i
]);
478 case GOMP_MAP_FORCE_FROM
:
480 acc_copyout_finalize (hostaddrs
[i
], sizes
[i
]);
482 acc_copyout (hostaddrs
[i
], sizes
[i
]);
485 gomp_fatal (">>>> GOACC_enter_exit_data UNHANDLED kind 0x%.2x",
492 bool copyfrom
= (kind
== GOMP_MAP_FORCE_FROM
493 || kind
== GOMP_MAP_FROM
);
494 gomp_acc_remove_pointer (hostaddrs
[i
], sizes
[i
], copyfrom
, async
,
496 /* See the above comment. */
501 acc_dev
->openacc
.async_set_async_func (acc_async_sync
);
505 goacc_wait (int async
, int num_waits
, va_list *ap
)
507 struct goacc_thread
*thr
= goacc_thread ();
508 struct gomp_device_descr
*acc_dev
= thr
->dev
;
512 int qid
= va_arg (*ap
, int);
514 if (acc_async_test (qid
))
517 if (async
== acc_async_sync
)
519 else if (qid
== async
)
520 ;/* If we're waiting on the same asynchronous queue as we're
521 launching on, the queue itself will order work as
522 required, so there's no need to wait explicitly. */
524 acc_dev
->openacc
.async_wait_async_func (qid
, async
);
529 GOACC_update (int device
, size_t mapnum
,
530 void **hostaddrs
, size_t *sizes
, unsigned short *kinds
,
531 int async
, int num_waits
, ...)
533 bool host_fallback
= device
== GOMP_DEVICE_HOST_FALLBACK
;
536 goacc_lazy_initialize ();
538 struct goacc_thread
*thr
= goacc_thread ();
539 struct gomp_device_descr
*acc_dev
= thr
->dev
;
541 if ((acc_dev
->capabilities
& GOMP_OFFLOAD_CAP_SHARED_MEM
)
549 va_start (ap
, num_waits
);
550 goacc_wait (async
, num_waits
, &ap
);
554 acc_dev
->openacc
.async_set_async_func (async
);
556 bool update_device
= false;
557 for (i
= 0; i
< mapnum
; ++i
)
559 unsigned char kind
= kinds
[i
] & 0xff;
563 case GOMP_MAP_POINTER
:
564 case GOMP_MAP_TO_PSET
:
567 case GOMP_MAP_ALWAYS_POINTER
:
570 /* Save the contents of the host pointer. */
571 void *dptr
= acc_deviceptr (hostaddrs
[i
-1]);
572 uintptr_t t
= *(uintptr_t *) hostaddrs
[i
];
574 /* Update the contents of the host pointer to reflect
575 the value of the allocated device memory in the
577 *(uintptr_t *) hostaddrs
[i
] = (uintptr_t)dptr
;
578 acc_update_device (hostaddrs
[i
], sizeof (uintptr_t));
580 /* Restore the host pointer. */
581 *(uintptr_t *) hostaddrs
[i
] = t
;
582 update_device
= false;
587 if (!acc_is_present (hostaddrs
[i
], sizes
[i
]))
589 update_device
= false;
593 case GOMP_MAP_FORCE_TO
:
594 update_device
= true;
595 acc_update_device (hostaddrs
[i
], sizes
[i
]);
599 if (!acc_is_present (hostaddrs
[i
], sizes
[i
]))
601 update_device
= false;
605 case GOMP_MAP_FORCE_FROM
:
606 update_device
= false;
607 acc_update_self (hostaddrs
[i
], sizes
[i
]);
611 gomp_fatal (">>>> GOACC_update UNHANDLED kind 0x%.2x", kind
);
616 acc_dev
->openacc
.async_set_async_func (acc_async_sync
);
620 GOACC_wait (int async
, int num_waits
, ...)
626 va_start (ap
, num_waits
);
627 goacc_wait (async
, num_waits
, &ap
);
630 else if (async
== acc_async_sync
)
632 else if (async
== acc_async_noval
)
633 goacc_thread ()->dev
->openacc
.async_wait_all_async_func (acc_async_noval
);
637 GOACC_get_num_threads (void)
643 GOACC_get_thread_num (void)
649 GOACC_declare (int device
, size_t mapnum
,
650 void **hostaddrs
, size_t *sizes
, unsigned short *kinds
)
654 for (i
= 0; i
< mapnum
; i
++)
656 unsigned char kind
= kinds
[i
] & 0xff;
658 if (kind
== GOMP_MAP_POINTER
|| kind
== GOMP_MAP_TO_PSET
)
663 case GOMP_MAP_FORCE_ALLOC
:
664 case GOMP_MAP_FORCE_FROM
:
665 case GOMP_MAP_FORCE_TO
:
666 case GOMP_MAP_POINTER
:
667 case GOMP_MAP_RELEASE
:
668 case GOMP_MAP_DELETE
:
669 GOACC_enter_exit_data (device
, 1, &hostaddrs
[i
], &sizes
[i
],
670 &kinds
[i
], GOMP_ASYNC_SYNC
, 0);
673 case GOMP_MAP_FORCE_DEVICEPTR
:
677 if (!acc_is_present (hostaddrs
[i
], sizes
[i
]))
678 GOACC_enter_exit_data (device
, 1, &hostaddrs
[i
], &sizes
[i
],
679 &kinds
[i
], GOMP_ASYNC_SYNC
, 0);
683 GOACC_enter_exit_data (device
, 1, &hostaddrs
[i
], &sizes
[i
],
684 &kinds
[i
], GOMP_ASYNC_SYNC
, 0);
689 GOACC_enter_exit_data (device
, 1, &hostaddrs
[i
], &sizes
[i
],
690 &kinds
[i
], GOMP_ASYNC_SYNC
, 0);
693 case GOMP_MAP_FORCE_PRESENT
:
694 if (!acc_is_present (hostaddrs
[i
], sizes
[i
]))
695 gomp_fatal ("[%p,%ld] is not mapped", hostaddrs
[i
],
696 (unsigned long) sizes
[i
]);