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