| 1 | // clang-format off |
|---|
| 2 | // RUN: %libomptarget-compilexx-generic && env LIBOMPTARGET_REUSE_BLOCKS_FOR_HIGH_TRIP_COUNT=False %libomptarget-run-generic 2>&1 | %fcheck-generic |
|---|
| 3 | // RUN: %libomptarget-compilexx-generic && %libomptarget-run-generic 2>&1 | %fcheck-generic --check-prefix=DEFAULT |
|---|
| 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 | // clang-format on |
|---|
| 12 | |
|---|
| 13 | /* |
|---|
| 14 | Check if there is a thread for each loop iteration |
|---|
| 15 | */ |
|---|
| 16 | #include <omp.h> |
|---|
| 17 | #include <stdio.h> |
|---|
| 18 | |
|---|
| 19 | int main() { |
|---|
| 20 | int N = 819200; |
|---|
| 21 | int num_threads[N]; |
|---|
| 22 | |
|---|
| 23 | #pragma omp target teams distribute parallel for |
|---|
| 24 | for (int j = 0; j < N; j++) { |
|---|
| 25 | num_threads[j] = omp_get_num_threads() * omp_get_num_teams(); |
|---|
| 26 | } |
|---|
| 27 | |
|---|
| 28 | if (num_threads[0] == N) |
|---|
| 29 | // CHECK: PASS |
|---|
| 30 | printf(format: "PASS\n" ); |
|---|
| 31 | else |
|---|
| 32 | // DEFAULT: FAIL |
|---|
| 33 | printf(format: "FAIL: num_threads: %d\n != N: %d" , num_threads[0], N); |
|---|
| 34 | return 0; |
|---|
| 35 | } |
|---|
| 36 | |
|---|
