openmp/runtime/test/misc_bugs/teams-reduction.c

   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:
  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
  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 }