1 #include <cassert>
2 #include <cstdint>
3 #include <iostream>
4 #include <string>
5
6 extern "C" {
7 struct ident_t;
8
9 using kmp_int32 = int32_t;
10 using kmp_int64 = int64_t;
11 using kmp_routine_entry_t = kmp_int32 (*)(kmp_int32, void *);
12 using kmp_intptr_t = intptr_t;
13
14 typedef struct kmp_depend_info {
15 kmp_intptr_t base_addr;
16 size_t len;
17 union {
18 unsigned char flag;
19 struct {
20 #if defined(__BYTE_ORDER__) && (__BYTE_ORDER__ == __ORDER_BIG_ENDIAN__)
21 unsigned all : 1;
22 unsigned unused : 3;
23 unsigned set : 1;
24 unsigned mtx : 1;
25 unsigned out : 1;
26 unsigned in : 1;
27 #else
28 unsigned in : 1;
29 unsigned out : 1;
30 unsigned mtx : 1;
31 unsigned set : 1;
32 unsigned unused : 3;
33 unsigned all : 1;
34 #endif
35 } flags;
36 };
37 } kmp_depend_info_t;
38
39 typedef union kmp_cmplrdata {
40 kmp_int32 priority;
41 kmp_routine_entry_t destructors;
42 } kmp_cmplrdata_t;
43
44 typedef struct kmp_task {
45 void *shareds;
46 kmp_routine_entry_t routine;
47 kmp_int32 part_id;
48 kmp_cmplrdata_t data1;
49 kmp_cmplrdata_t data2;
50 } kmp_task_t;
51
52 int32_t __kmpc_global_thread_num(void *);
53 kmp_task_t *__kmpc_omp_task_alloc(ident_t *, kmp_int32, kmp_int32, size_t,
54 size_t, kmp_routine_entry_t);
55 kmp_task_t *__kmpc_omp_target_task_alloc(ident_t *, kmp_int32, kmp_int32,
56 size_t, size_t, kmp_routine_entry_t,
57 kmp_int64);
58 kmp_int32 __kmpc_omp_taskwait(ident_t *, kmp_int32);
59 kmp_int32 __kmpc_omp_task(ident_t *, kmp_int32, kmp_task_t *);
60 kmp_int32 __kmpc_omp_task_with_deps(ident_t *loc_ref, kmp_int32 gtid,
61 kmp_task_t *new_task, kmp_int32 ndeps,
62 kmp_depend_info_t *dep_list,
63 kmp_int32 ndeps_noalias,
64 kmp_depend_info_t *noalias_dep_list);
65 void __kmpc_taskgroup(ident_t *, kmp_int32);
66 void __kmpc_end_taskgroup(ident_t *, kmp_int32);
67 }
68
get_num_hidden_helper_threads()69 static kmp_int32 get_num_hidden_helper_threads() {
70 static kmp_int32 __kmp_hidden_helper_threads_num = 8;
71 if (const char *env = std::getenv("LIBOMP_NUM_HIDDEN_HELPER_THREADS")) {
72 return std::stoi(env);
73 }
74 return __kmp_hidden_helper_threads_num;
75 }
76