xref: /llvm-project/offload/test/offloading/schedule.c (revision 1a0cf245ac86c2f35c89cab47f83e9b474032e41)
1 // clang-format off
2 // RUN: %libomptarget-compile-generic && %libomptarget-run-generic 2>&1 | %fcheck-generic
3 // clang-format on
4 
5 // UNSUPPORTED: aarch64-unknown-linux-gnu
6 // UNSUPPORTED: aarch64-unknown-linux-gnu-LTO
7 // UNSUPPORTED: x86_64-unknown-linux-gnu
8 // UNSUPPORTED: x86_64-unknown-linux-gnu-LTO
9 // UNSUPPORTED: s390x-ibm-linux-gnu
10 // UNSUPPORTED: s390x-ibm-linux-gnu-LTO
11 
12 // REQUIRES: amdgcn-amd-amdhsa
13 
14 #include <omp.h>
15 #include <stdio.h>
16 #include <stdlib.h>
17 
18 int ordered_example(int lb, int ub, int stride, int nteams) {
19   int i;
20   int size = (ub - lb) / stride;
21   double *output = (double *)malloc(size * sizeof(double));
22 
23 #pragma omp target teams map(from : output[0 : size]) num_teams(nteams)        \
24     thread_limit(128)
25 #pragma omp parallel for ordered schedule(dynamic)
26   for (i = lb; i < ub; i += stride) {
27 #pragma omp ordered
28     { output[(i - lb) / stride] = omp_get_wtime(); }
29   }
30 
31   // verification
32   for (int j = 0; j < size; j++) {
33     for (int jj = j + 1; jj < size; jj++) {
34       if (output[j] > output[jj]) {
35         printf("Fail to schedule in order.\n");
36         free(output);
37         return 1;
38       }
39     }
40   }
41 
42   free(output);
43 
44   printf("test ordered OK\n");
45 
46   return 0;
47 }
48 
49 int NO_order_example(int lb, int ub, int stride, int nteams) {
50   int i;
51   int size = (ub - lb) / stride;
52   double *output = (double *)malloc(size * sizeof(double));
53 
54 #pragma omp target teams map(from : output[0 : size]) num_teams(nteams)        \
55     thread_limit(128)
56 #pragma omp parallel for schedule(dynamic)
57   for (i = lb; i < ub; i += stride) {
58     output[(i - lb) / stride] = omp_get_wtime();
59   }
60 
61   // verification
62   for (int j = 0; j < size; j++) {
63     for (int jj = j + 1; jj < size; jj++) {
64       if (output[j] > output[jj]) {
65         printf("Fail to schedule in order.\n");
66         free(output);
67         return 1;
68       }
69     }
70   }
71 
72   free(output);
73 
74   printf("test no order OK\n");
75 
76   return 0;
77 }
78 
79 int main() {
80   // CHECK: test no order OK
81   NO_order_example(0, 10, 1, 8);
82   // CHECK: test ordered OK
83   return ordered_example(0, 10, 1, 8);
84 }
85