94 lines · cpp
1// RUN: %libomp-cxx-compile-and-run2 3#include <stdio.h>4#include <omp.h>5 6#define NT 47#define INIT 108 9/*10The test emulates code generation needed for reduction with task modifier on11parallel construct.12 13Note: tasks could just use in_reduction clause, but compiler does not accept14this because of bug: it mistakenly requires reduction item to be shared, which15is only true for reduction on worksharing and wrong for task reductions.16*/17 18//------------------------------------------------19// OpenMP runtime library routines20#ifdef __cplusplus21extern "C" {22#endif23extern 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 __cplusplus29}30#endif31 32//------------------------------------------------33// Compiler-generated code34 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 0, // 1 - worksharing construct, 0 - parallel66 1, // number of reduction objects67 &r_var // related data68 );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, &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, &var);81 *p_var += 1;82 }83 }84 __kmpc_task_reduction_modifier_fini(NULL, gtid, 0);85 }86 if (var == INIT + NT * 3 - 1) {87 printf("passed\n");88 return 0;89 } else {90 printf("failed: var = %d (!= %d)\n", var, INIT + NT * 3 - 1);91 return 1;92 }93}94