1// clang-format off
2// RUN: %libomptarget-compilexx-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
17#define N 100
18
19bool schedule(int lb, int ub, int stride, int chunk) {
20 int i;
21
22 int result[N];
23 for (i = 0; i < N; i++) {
24 result[i] = 0;
25 }
26
27#pragma omp target map(tofrom : result[ : N])
28 {
29 int a = 0;
30#pragma omp parallel for schedule(dynamic, chunk)
31 for (i = lb; i < ub; i += stride) {
32 result[i] += i + a;
33 }
34 }
35
36 int value = 0;
37 bool success = true;
38 for (i = 0; i < N; i += stride) {
39 if (value != result[i]) {
40 printf(format: "ERROR: result[%d] = %d instead of %d\n", i, result[i], value);
41 success = false;
42 break;
43 }
44 value += stride;
45 }
46
47 return success;
48}
49
50int main() {
51 // CHECK: SUCCESS CHUNK SIZE 1
52 if (schedule(lb: 0, N, stride: 5, chunk: 1))
53 printf(format: "SUCCESS CHUNK SIZE 1\n");
54
55 // CHECK: SUCCESS CHUNK SIZE 3
56 if (schedule(lb: 0, N, stride: 5, chunk: 3))
57 printf(format: "SUCCESS CHUNK SIZE 3\n");
58
59 return 0;
60}
61

source code of offload/test/offloading/dynamic-schedule-non-spmd.cpp