LLVM OpenMP* Runtime Library
Loading...
Searching...
No Matches
kmp_invoke_microtask.cpp
1#include "kmp.h"
2
3#if !(KMP_ARCH_X86 || KMP_ARCH_X86_64 || KMP_MIC || KMP_ARCH_AARCH64 || \
4 KMP_ARCH_PPC64 || KMP_ARCH_RISCV64 || KMP_ARCH_LOONGARCH64 || \
5 KMP_ARCH_ARM || KMP_ARCH_VE || KMP_ARCH_S390X || KMP_ARCH_PPC_XCOFF || \
6 KMP_ARCH_AARCH64_32)
7
8// Because WebAssembly will use `call_indirect` to invoke the microtask and
9// WebAssembly indirect calls check that the called signature is a precise
10// match, we need to cast each microtask function pointer back from `void *` to
11// its original type.
12typedef void (*microtask_t0)(int *, int *);
13typedef void (*microtask_t1)(int *, int *, void *);
14typedef void (*microtask_t2)(int *, int *, void *, void *);
15typedef void (*microtask_t3)(int *, int *, void *, void *, void *);
16typedef void (*microtask_t4)(int *, int *, void *, void *, void *, void *);
17typedef void (*microtask_t5)(int *, int *, void *, void *, void *, void *,
18 void *);
19typedef void (*microtask_t6)(int *, int *, void *, void *, void *, void *,
20 void *, void *);
21typedef void (*microtask_t7)(int *, int *, void *, void *, void *, void *,
22 void *, void *, void *);
23typedef void (*microtask_t8)(int *, int *, void *, void *, void *, void *,
24 void *, void *, void *, void *);
25typedef void (*microtask_t9)(int *, int *, void *, void *, void *, void *,
26 void *, void *, void *, void *, void *);
27typedef void (*microtask_t10)(int *, int *, void *, void *, void *, void *,
28 void *, void *, void *, void *, void *, void *);
29typedef void (*microtask_t11)(int *, int *, void *, void *, void *, void *,
30 void *, void *, void *, void *, void *, void *,
31 void *);
32typedef void (*microtask_t12)(int *, int *, void *, void *, void *, void *,
33 void *, void *, void *, void *, void *, void *,
34 void *, void *);
35typedef void (*microtask_t13)(int *, int *, void *, void *, void *, void *,
36 void *, void *, void *, void *, void *, void *,
37 void *, void *, void *);
38typedef void (*microtask_t14)(int *, int *, void *, void *, void *, void *,
39 void *, void *, void *, void *, void *, void *,
40 void *, void *, void *, void *);
41typedef void (*microtask_t15)(int *, int *, void *, void *, void *, void *,
42 void *, void *, void *, void *, void *, void *,
43 void *, void *, void *, void *, void *);
44
45// we really only need the case with 1 argument, because CLANG always build
46// a struct of pointers to shared variables referenced in the outlined function
47int __kmp_invoke_microtask(microtask_t pkfn, int gtid, int tid, int argc,
48 void *p_argv[]
49#if OMPT_SUPPORT
50 ,
51 void **exit_frame_ptr
52#endif
53) {
54#if OMPT_SUPPORT
55 *exit_frame_ptr = OMPT_GET_FRAME_ADDRESS(0);
56#endif
57
58 switch (argc) {
59 default:
60 fprintf(stderr, "Too many args to microtask: %d!\n", argc);
61 fflush(stderr);
62 exit(-1);
63 case 0:
64 (*(microtask_t0)pkfn)(&gtid, &tid);
65 break;
66 case 1:
67 (*(microtask_t1)pkfn)(&gtid, &tid, p_argv[0]);
68 break;
69 case 2:
70 (*(microtask_t2)pkfn)(&gtid, &tid, p_argv[0], p_argv[1]);
71 break;
72 case 3:
73 (*(microtask_t3)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2]);
74 break;
75 case 4:
76 (*(microtask_t4)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
77 p_argv[3]);
78 break;
79 case 5:
80 (*(microtask_t5)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
81 p_argv[3], p_argv[4]);
82 break;
83 case 6:
84 (*(microtask_t6)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
85 p_argv[3], p_argv[4], p_argv[5]);
86 break;
87 case 7:
88 (*(microtask_t7)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
89 p_argv[3], p_argv[4], p_argv[5], p_argv[6]);
90 break;
91 case 8:
92 (*(microtask_t8)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
93 p_argv[3], p_argv[4], p_argv[5], p_argv[6],
94 p_argv[7]);
95 break;
96 case 9:
97 (*(microtask_t9)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
98 p_argv[3], p_argv[4], p_argv[5], p_argv[6], p_argv[7],
99 p_argv[8]);
100 break;
101 case 10:
102 (*(microtask_t10)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
103 p_argv[3], p_argv[4], p_argv[5], p_argv[6],
104 p_argv[7], p_argv[8], p_argv[9]);
105 break;
106 case 11:
107 (*(microtask_t11)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
108 p_argv[3], p_argv[4], p_argv[5], p_argv[6],
109 p_argv[7], p_argv[8], p_argv[9], p_argv[10]);
110 break;
111 case 12:
112 (*(microtask_t12)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
113 p_argv[3], p_argv[4], p_argv[5], p_argv[6],
114 p_argv[7], p_argv[8], p_argv[9], p_argv[10],
115 p_argv[11]);
116 break;
117 case 13:
118 (*(microtask_t13)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
119 p_argv[3], p_argv[4], p_argv[5], p_argv[6],
120 p_argv[7], p_argv[8], p_argv[9], p_argv[10],
121 p_argv[11], p_argv[12]);
122 break;
123 case 14:
124 (*(microtask_t14)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
125 p_argv[3], p_argv[4], p_argv[5], p_argv[6],
126 p_argv[7], p_argv[8], p_argv[9], p_argv[10],
127 p_argv[11], p_argv[12], p_argv[13]);
128 break;
129 case 15:
130 (*(microtask_t15)pkfn)(&gtid, &tid, p_argv[0], p_argv[1], p_argv[2],
131 p_argv[3], p_argv[4], p_argv[5], p_argv[6],
132 p_argv[7], p_argv[8], p_argv[9], p_argv[10],
133 p_argv[11], p_argv[12], p_argv[13], p_argv[14]);
134 break;
135 }
136
137 return 1;
138}
139
140#endif
Generated by doxygen 1.15.0