libgomp/testsuite/libgomp.c-c++-common/taskloop-reduction-4.c

/* { dg-do run } */
/* { dg-options "-O2" } */
/* { dg-additional-options "-std=c99" { target c } } */
/* { dg-additional-options "-msse2" { target sse2_runtime } } */
/* { dg-additional-options "-mavx" { target avx_runtime } } */

#include <omp.h>
#include <stdlib.h>

#define N 1024
long int u[N], m, n, o;

__attribute__((noipa)) void
foo (void)
{
  int i = -1;
  #pragma omp master taskloop simd reduction (+:m) grainsize (64)
  for (i = 0; i < N; ++i)
    m += u[i];
  if (i != (omp_get_thread_num () ? -1 : N))
    abort ();
}

__attribute__((noipa)) void
bar (int x)
{
  int i = -1;
  #pragma omp master taskloop simd in_reduction (+:n) grainsize (64)
  for (i = (x & 1) * (N / 2); i < (x & 1) * (N / 2) + (N / 2); i++)
    n += 2 * u[i];
  if (i != (omp_get_thread_num () ? -1 : (x & 1) * (N / 2) + (N / 2)))
    abort ();
}

__attribute__((noipa)) void
baz (void)
{
  int i;
  #pragma omp parallel master taskloop simd reduction (+:o) grainsize (64)
  for (i = 0; i < N; ++i)
    o += u[i];
  if (i != N)
    abort ();
}

int
main ()
{
  int i;
  for (i = 0; i < N; ++i)
    u[i] = i;
  #pragma omp parallel
  {
    foo ();
    #pragma omp taskgroup task_reduction (+:n)
    {
      bar (0);
      bar (1);
    }
  }
  baz ();
  if (m != (long)(N - 1) * (N / 2) || n != (long)(N - 1) * N || o != m)
    abort ();
  return 0;
}