| 1 | // RUN: %libomptarget-compile-generic && env LIBOMPTARGET_DEBUG=1 %libomptarget-run-generic 2>&1 | %fcheck-generic -allow-empty -check-prefix=DEBUG |
|---|
| 2 | // REQUIRES: libomptarget-debug |
|---|
| 3 | |
|---|
| 4 | /* |
|---|
| 5 | Test for looptripcount being popped from runtime stack. |
|---|
| 6 | */ |
|---|
| 7 | #include <omp.h> |
|---|
| 8 | #include <stdio.h> |
|---|
| 9 | int main() { |
|---|
| 10 | int N = 128; |
|---|
| 11 | int NN = 1024; |
|---|
| 12 | int num_teams[NN]; |
|---|
| 13 | int num_threads[NN]; |
|---|
| 14 | |
|---|
| 15 | printf(format: "#pragma omp target teams distribute parallel for thread_limit(4)\n" ); |
|---|
| 16 | #pragma omp target teams distribute parallel for thread_limit(4) |
|---|
| 17 | for (int j = 0; j < N; j++) { |
|---|
| 18 | num_threads[j] = omp_get_num_threads(); |
|---|
| 19 | num_teams[j] = omp_get_num_teams(); |
|---|
| 20 | } |
|---|
| 21 | printf(format: "num_threads %d num_teams %d\n" , num_threads[0], num_teams[0]); |
|---|
| 22 | // DEBUG: loop trip count is 128 |
|---|
| 23 | printf(format: "#pragma omp target teams distribute parallel for\n" ); |
|---|
| 24 | #pragma omp target teams distribute parallel for |
|---|
| 25 | for (int j = 0; j < N; j++) { |
|---|
| 26 | num_threads[j] = omp_get_num_threads(); |
|---|
| 27 | num_teams[j] = omp_get_num_teams(); |
|---|
| 28 | } |
|---|
| 29 | printf(format: "num_threads %d num_teams %d\n" , num_threads[0], num_teams[0]); |
|---|
| 30 | // DEBUG: loop trip count is 128 |
|---|
| 31 | return 0; |
|---|
| 32 | } |
|---|
| 33 | |
|---|
