xref: /llvm-project/openmp/runtime/test/tasking/omp_record_replay.cpp (revision 36d4e4c9b5f6cd0577b6029055b825caaec2dd11)
1*36d4e4c9SChenle Yu // REQUIRES: ompx_taskgraph
2*36d4e4c9SChenle Yu // RUN: %libomp-cxx-compile-and-run
3*36d4e4c9SChenle Yu #include <iostream>
4*36d4e4c9SChenle Yu #include <cassert>
5*36d4e4c9SChenle Yu #define NT 100
6*36d4e4c9SChenle Yu 
7*36d4e4c9SChenle Yu // Compiler-generated code (emulation)
8*36d4e4c9SChenle Yu typedef struct ident {
9*36d4e4c9SChenle Yu     void* dummy;
10*36d4e4c9SChenle Yu } ident_t;
11*36d4e4c9SChenle Yu 
12*36d4e4c9SChenle Yu 
13*36d4e4c9SChenle Yu #ifdef __cplusplus
14*36d4e4c9SChenle Yu extern "C" {
15*36d4e4c9SChenle Yu   int __kmpc_global_thread_num(ident_t *);
16*36d4e4c9SChenle Yu   int __kmpc_start_record_task(ident_t *, int, int, int);
17*36d4e4c9SChenle Yu   void __kmpc_end_record_task(ident_t *, int, int , int);
18*36d4e4c9SChenle Yu }
19*36d4e4c9SChenle Yu #endif
20*36d4e4c9SChenle Yu 
func(int * num_exec)21*36d4e4c9SChenle Yu void func(int *num_exec) {
22*36d4e4c9SChenle Yu   (*num_exec)++;
23*36d4e4c9SChenle Yu }
24*36d4e4c9SChenle Yu 
main()25*36d4e4c9SChenle Yu int main() {
26*36d4e4c9SChenle Yu   int num_exec = 0;
27*36d4e4c9SChenle Yu   int num_tasks = 0;
28*36d4e4c9SChenle Yu   int x=0;
29*36d4e4c9SChenle Yu   #pragma omp parallel
30*36d4e4c9SChenle Yu   #pragma omp single
31*36d4e4c9SChenle Yu   for (int iter = 0; iter < NT; ++iter) {
32*36d4e4c9SChenle Yu     int gtid = __kmpc_global_thread_num(nullptr);
33*36d4e4c9SChenle Yu     int res =  __kmpc_start_record_task(nullptr, gtid, /* kmp_tdg_flags */ 0, /* tdg_id */0);
34*36d4e4c9SChenle Yu     if (res) {
35*36d4e4c9SChenle Yu       num_tasks++;
36*36d4e4c9SChenle Yu       #pragma omp task
37*36d4e4c9SChenle Yu       func(&num_exec);
38*36d4e4c9SChenle Yu     }
39*36d4e4c9SChenle Yu     __kmpc_end_record_task(nullptr, gtid, /* kmp_tdg_flags */0, /* tdg_id */0);
40*36d4e4c9SChenle Yu   }
41*36d4e4c9SChenle Yu 
42*36d4e4c9SChenle Yu   assert(num_tasks==1);
43*36d4e4c9SChenle Yu   assert(num_exec==NT);
44*36d4e4c9SChenle Yu 
45*36d4e4c9SChenle Yu   std::cout << "Passed" << std::endl;
46*36d4e4c9SChenle Yu   return 0;
47*36d4e4c9SChenle Yu }
48*36d4e4c9SChenle Yu // CHECK: Passed
49