1 // RUN: %libomp-compile-and-run
2 //
3 // The test checks the teams construct with reduction executed on the host.
4 //
5
6 #include <stdio.h>
7 #include <omp.h>
8
9 #include <stdint.h>
10
11 #ifndef N_TEAMS
12 #define N_TEAMS 4
13 #endif
14 #ifndef N_THR
15 #define N_THR 3
16 #endif
17
18 // Internal library stuff to emulate compiler's code generation:
19 #ifdef __cplusplus
20 extern "C" {
21 #endif
22
23 typedef struct {
24 int32_t reserved_1;
25 int32_t flags;
26 int32_t reserved_2;
27 int32_t reserved_3;
28 char const *psource;
29 } ident_t;
30
31 static ident_t dummy_loc = {0, 2, 0, 0, ";dummyFile;dummyFunc;0;0;;"};
32
33 typedef union {
34 // The global will be used as pointer, so we need to make sure that the
35 // compiler correctly aligns the global...
36 void *ptr;
37 int32_t data[8];
38 } kmp_critical_name;
39 kmp_critical_name crit;
40
41 int32_t __kmpc_global_thread_num(ident_t *);
42 void __kmpc_push_num_teams(ident_t *, int32_t global_tid, int32_t num_teams,
43 int32_t num_threads);
44 void __kmpc_fork_teams(ident_t *, int32_t argc, void *microtask, ...);
45 int32_t __kmpc_reduce(ident_t *, int32_t global_tid, int32_t num_vars,
46 size_t reduce_size, void *reduce_data, void *reduce_func,
47 kmp_critical_name *lck);
48 void __kmpc_end_reduce(ident_t *, int32_t global_tid, kmp_critical_name *lck);
49
50 #ifdef __cplusplus
51 }
52 #endif
53
54 // Outlined entry point:
outlined(int32_t * gtid,int32_t * tid)55 void outlined(int32_t *gtid, int32_t *tid) {
56 int32_t ret = __kmpc_reduce(&dummy_loc, *gtid, 0, 0, NULL, NULL, &crit);
57 __kmpc_end_reduce(&dummy_loc, *gtid, &crit);
58 }
59
main()60 int main() {
61 int32_t th = __kmpc_global_thread_num(NULL); // registers initial thread
62 __kmpc_push_num_teams(&dummy_loc, th, N_TEAMS, N_THR);
63 __kmpc_fork_teams(&dummy_loc, 0, &outlined);
64
65 // Test did not hang -> passed!
66 printf("passed\n");
67 return 0;
68 }
69