1// RUN: %libomp-cxx-compile-and-run
2
3#include <stdio.h>
4#include <omp.h>
5
6#define NT 4
7#define INIT 10
8
9/*
10The test emulates code generation needed for reduction with task modifier on
11parallel construct.
12
13Note: tasks could just use in_reduction clause, but compiler does not accept
14this because of bug: it mistakenly requires reduction item to be shared, which
15is only true for reduction on worksharing and wrong for task reductions.
16*/
17
18//------------------------------------------------
19// OpenMP runtime library routines
20#ifdef __cplusplus
21extern "C" {
22#endif
23extern void *__kmpc_task_reduction_get_th_data(int gtid, void *tg, void *item);
24extern void *__kmpc_task_reduction_modifier_init(void *loc, int gtid, int is_ws,
25 int num, void *data);
26extern void __kmpc_task_reduction_modifier_fini(void *loc, int gtid, int is_ws);
27extern int __kmpc_global_thread_num(void *);
28#ifdef __cplusplus
29}
30#endif
31
32//------------------------------------------------
33// Compiler-generated code
34
35typedef struct red_input {
36 void *reduce_shar; /**< shared between tasks item to reduce into */
37 size_t reduce_size; /**< size of data item in bytes */
38 // three compiler-generated routines (init, fini are optional):
39 void *reduce_init; /**< data initialization routine (single parameter) */
40 void *reduce_fini; /**< data finalization routine */
41 void *reduce_comb; /**< data combiner routine */
42 unsigned flags; /**< flags for additional info from compiler */
43} red_input_t;
44
45void i_comb(void *lhs, void *rhs) { *(int *)lhs += *(int *)rhs; }
46
47int main() {
48 int var = INIT;
49 omp_set_dynamic(0);
50 omp_set_num_threads(NT);
51// #pragma omp parallel reduction(task,+:var)
52#pragma omp parallel reduction(+ : var)
53 {
54 int gtid = __kmpc_global_thread_num(NULL);
55 void *tg; // pointer to taskgroup (optional)
56 red_input_t r_var;
57 r_var.reduce_shar = &var;
58 r_var.reduce_size = sizeof(var);
59 r_var.reduce_init = NULL;
60 r_var.reduce_fini = NULL;
61 r_var.reduce_comb = (void *)&i_comb;
62 tg = __kmpc_task_reduction_modifier_init(
63 NULL, // ident_t loc;
64 gtid,
65 is_ws: 0, // 1 - worksharing construct, 0 - parallel
66 num: 1, // number of reduction objects
67 data: &r_var // related data
68 );
69 var++;
70#pragma omp task /*in_reduction(+:var)*/ shared(var)
71 {
72 int gtid = __kmpc_global_thread_num(NULL);
73 int *p_var = (int *)__kmpc_task_reduction_get_th_data(gtid, tg, item: &var);
74 *p_var += 1;
75 }
76 if (omp_get_thread_num() > 0) {
77#pragma omp task /*in_reduction(+:var)*/ shared(var)
78 {
79 int gtid = __kmpc_global_thread_num(NULL);
80 int *p_var = (int *)__kmpc_task_reduction_get_th_data(gtid, tg, item: &var);
81 *p_var += 1;
82 }
83 }
84 __kmpc_task_reduction_modifier_fini(NULL, gtid, is_ws: 0);
85 }
86 if (var == INIT + NT * 3 - 1) {
87 printf(format: "passed\n");
88 return 0;
89 } else {
90 printf(format: "failed: var = %d (!= %d)\n", var, INIT + NT * 3 - 1);
91 return 1;
92 }
93}
94

source code of openmp/runtime/test/tasking/kmp_task_modifier_simple_par_old.cpp