File: teams-distr-on-host.c

package info (click to toggle)
llvm-toolchain-16 1%3A16.0.6-15~deb11u2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 1,634,820 kB
  • sloc: cpp: 6,179,261; ansic: 1,216,205; asm: 741,319; python: 196,614; objc: 75,325; f90: 49,640; lisp: 32,396; pascal: 12,286; sh: 9,394; perl: 7,442; ml: 5,494; awk: 3,523; makefile: 2,723; javascript: 1,206; xml: 886; fortran: 581; cs: 573
file content (46 lines) | stat: -rw-r--r-- 989 bytes parent folder | download | duplicates (3) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
// The test supposes no offload, pure host execution.
// It checks that the bug in implementation of distribute construct is fixed.

// RUN: %libomp-compile-and-run
// UNSUPPORTED: icc

#include <stdio.h>
#include <omp.h>

int main()
{
  const int size = 4;
  int wrong_counts = 0;
  omp_set_num_threads(2);
  #pragma omp parallel reduction(+:wrong_counts)
  {
    int i;
    int A[size];
    int th = omp_get_thread_num();
    for(i = 0; i < size; i++)
      A[i] = 0;

    #pragma omp target teams distribute map(tofrom: A[:size]) private(i)
    for(i = 0; i < size; i++)
    {
      A[i] = i;
      printf("th %d, team %d, i %d\n", th, omp_get_team_num(), i);
    }
    #pragma omp critical
    {
      printf("tid = %d\n", th);
      for(i = 0; i < size; i++)
      {
        if (A[i] != i) wrong_counts++;
        printf("  %d", A[i]);
      }
      printf("\n");
    }
  }
  if (wrong_counts) {
    printf("failed\n");
  } else {
    printf("passed\n");
  }
  return wrong_counts;
}