63 lines
1.9 KiB
C++
63 lines
1.9 KiB
C++
#include <cassert>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
extern "C" {
|
|
struct ident_t;
|
|
|
|
using kmp_int32 = int32_t;
|
|
using kmp_int64 = int64_t;
|
|
using kmp_routine_entry_t = kmp_int32 (*)(kmp_int32, void *);
|
|
using kmp_intptr_t = intptr_t;
|
|
|
|
typedef struct kmp_depend_info {
|
|
kmp_intptr_t base_addr;
|
|
size_t len;
|
|
union {
|
|
unsigned char flag;
|
|
struct {
|
|
bool in : 1;
|
|
bool out : 1;
|
|
bool mtx : 1;
|
|
} flags;
|
|
};
|
|
} kmp_depend_info_t;
|
|
|
|
typedef union kmp_cmplrdata {
|
|
kmp_int32 priority;
|
|
kmp_routine_entry_t destructors;
|
|
} kmp_cmplrdata_t;
|
|
|
|
typedef struct kmp_task {
|
|
void *shareds;
|
|
kmp_routine_entry_t routine;
|
|
kmp_int32 part_id;
|
|
kmp_cmplrdata_t data1;
|
|
kmp_cmplrdata_t data2;
|
|
} kmp_task_t;
|
|
|
|
int32_t __kmpc_global_thread_num(void *);
|
|
kmp_task_t *__kmpc_omp_task_alloc(ident_t *, kmp_int32, kmp_int32, size_t,
|
|
size_t, kmp_routine_entry_t);
|
|
kmp_task_t *__kmpc_omp_target_task_alloc(ident_t *, kmp_int32, kmp_int32,
|
|
size_t, size_t, kmp_routine_entry_t,
|
|
kmp_int64);
|
|
kmp_int32 __kmpc_omp_taskwait(ident_t *, kmp_int32);
|
|
kmp_int32 __kmpc_omp_task(ident_t *, kmp_int32, kmp_task_t *);
|
|
kmp_int32 __kmpc_omp_task_with_deps(ident_t *loc_ref, kmp_int32 gtid,
|
|
kmp_task_t *new_task, kmp_int32 ndeps,
|
|
kmp_depend_info_t *dep_list,
|
|
kmp_int32 ndeps_noalias,
|
|
kmp_depend_info_t *noalias_dep_list);
|
|
void __kmpc_taskgroup(ident_t *, kmp_int32);
|
|
void __kmpc_end_taskgroup(ident_t *, kmp_int32);
|
|
}
|
|
|
|
static kmp_int32 get_num_hidden_helper_threads() {
|
|
static kmp_int32 __kmp_hidden_helper_threads_num = 8;
|
|
if (const char *env = std::getenv("LIBOMP_NUM_HIDDEN_HELPER_THREADS")) {
|
|
return std::stoi(env);
|
|
}
|
|
return __kmp_hidden_helper_threads_num;
|
|
}
|