@@ -70,10 +70,28 @@ typedef struct kmp_task {
7070 int32_t part_id;
7171} kmp_task_t ;
7272
73- int32_t __kmpc_global_thread_num (void *) __attribute__((weak));
74- bool __kmpc_omp_has_task_team (int32_t gtid) __attribute__((weak));
75- void **__kmpc_omp_get_target_async_handle_ptr (int32_t gtid)
76- __attribute__((weak));
73+ // Implemented in libomp, they are called from within __tgt_* functions.
74+ int32_t __kmpc_global_thread_num (void *);
75+ bool __kmpc_omp_has_task_team (int32_t gtid);
76+ void **__kmpc_omp_get_target_async_handle_ptr (int32_t gtid);
77+ int __kmpc_get_target_offload (void );
78+ int __kmpc_get_target_offload (void );
79+ kmp_task_t *__kmpc_omp_task_alloc (ident_t *loc_ref, int32_t gtid, int32_t flags,
80+ size_t sizeof_kmp_task_t ,
81+ size_t sizeof_shareds,
82+ kmp_routine_entry_t task_entry);
83+ kmp_task_t *
84+ __kmpc_omp_target_task_alloc (ident_t *loc_ref, int32_t gtid, int32_t flags,
85+ size_t sizeof_kmp_task_t , size_t sizeof_shareds,
86+ kmp_routine_entry_t task_entry, int64_t device_id);
87+ int32_t __kmpc_omp_task_with_deps (ident_t *loc_ref, int32_t gtid,
88+ kmp_task_t *new_task, int32_t ndeps,
89+ kmp_depend_info_t *dep_list,
90+ int32_t ndeps_noalias,
91+ kmp_depend_info_t *noalias_dep_list);
92+ void __kmpc_omp_wait_deps (ident_t *loc_ref, int32_t gtid, int32_t ndeps,
93+ kmp_depend_info_t *dep_list, int32_t ndeps_noalias,
94+ kmp_depend_info_t *noalias_dep_list);
7795
7896/* *
7997 * The argument set that is passed from asynchronous memory copy to block
@@ -120,7 +138,7 @@ struct TargetMemcpyArgsTy {
120138 : Dst(Dst), Src(Src), DstDevice(DstDevice), SrcDevice(SrcDevice),
121139 IsRectMemcpy (false ), Length(Length), DstOffset(DstOffset),
122140 SrcOffset(SrcOffset), ElementSize(0 ), NumDims(0 ), Volume(0 ),
123- DstOffsets(0 ), SrcOffsets(0 ), DstDimensions(0 ), SrcDimensions(0 ){};
141+ DstOffsets(0 ), SrcOffsets(0 ), DstDimensions(0 ), SrcDimensions(0 ) {};
124142
125143 /* *
126144 * Constructor for rectangle dimensional copy
@@ -134,7 +152,7 @@ struct TargetMemcpyArgsTy {
134152 IsRectMemcpy(true ), Length(0 ), DstOffset(0 ), SrcOffset(0 ),
135153 ElementSize(ElementSize), NumDims(NumDims), Volume(Volume),
136154 DstOffsets(DstOffsets), SrcOffsets(SrcOffsets),
137- DstDimensions(DstDimensions), SrcDimensions(SrcDimensions){};
155+ DstDimensions(DstDimensions), SrcDimensions(SrcDimensions) {};
138156};
139157
140158struct TargetMemsetArgsTy {
0 commit comments