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