1 | // RUN: %libomp-compile-and-run |
2 | #include <stdio.h> |
3 | #include <stdlib.h> |
4 | #include <math.h> |
5 | #include "omp_testsuite.h" |
6 | |
7 | int sum1; |
8 | #pragma omp threadprivate(sum1) |
9 | |
10 | int test_omp_for_auto() |
11 | { |
12 | int j; |
13 | int sum; |
14 | int sum0; |
15 | int known_sum; |
16 | int threadsnum; |
17 | |
18 | sum = 0; |
19 | sum0 = 12345; |
20 | |
21 | // array which keeps track of which threads participated in the for loop |
22 | // e.g., given 4 threads, [ 0 | 1 | 1 | 0 ] implies |
23 | // threads 0 and 3 did not, threads 1 and 2 did |
24 | int max_threads = omp_get_max_threads(); |
25 | int* active_threads = (int*)malloc(size: sizeof(int)*max_threads); |
26 | for(j = 0; j < max_threads; j++) |
27 | active_threads[j] = 0; |
28 | |
29 | #pragma omp parallel |
30 | { |
31 | int i; |
32 | sum1 = 0; |
33 | #pragma omp for firstprivate(sum0) schedule(auto) |
34 | for (i = 1; i <= LOOPCOUNT; i++) { |
35 | active_threads[omp_get_thread_num()] = 1; |
36 | sum0 = sum0 + i; |
37 | sum1 = sum0; |
38 | } |
39 | |
40 | #pragma omp critical |
41 | { |
42 | sum = sum + sum1; |
43 | } |
44 | } |
45 | |
46 | // count the threads that participated (sum is stored in threadsnum) |
47 | threadsnum=0; |
48 | for(j = 0; j < max_threads; j++) { |
49 | if(active_threads[j]) |
50 | threadsnum++; |
51 | } |
52 | free(ptr: active_threads); |
53 | |
54 | known_sum = 12345 * threadsnum + (LOOPCOUNT * (LOOPCOUNT + 1)) / 2; |
55 | return (known_sum == sum); |
56 | } |
57 | |
58 | int main() |
59 | { |
60 | int i; |
61 | int num_failed=0; |
62 | |
63 | for(i = 0; i < REPETITIONS; i++) { |
64 | if(!test_omp_for_auto()) { |
65 | num_failed++; |
66 | } |
67 | } |
68 | return num_failed; |
69 | } |
70 | |