forked from OSchip/llvm-project
75 lines
1.8 KiB
C
75 lines
1.8 KiB
C
// RUN: %libomp-compile-and-run
|
|
#include <stdio.h>
|
|
#include <math.h>
|
|
#include "omp_testsuite.h"
|
|
#include "omp_my_sleep.h"
|
|
|
|
int test_omp_taskwait()
|
|
{
|
|
int result1 = 0; /* Stores number of not finished tasks after the taskwait */
|
|
int result2 = 0; /* Stores number of wrong array elements at the end */
|
|
int array[NUM_TASKS];
|
|
int i;
|
|
|
|
/* fill array */
|
|
for (i = 0; i < NUM_TASKS; i++)
|
|
array[i] = 0;
|
|
|
|
#pragma omp parallel
|
|
{
|
|
#pragma omp single
|
|
{
|
|
for (i = 0; i < NUM_TASKS; i++) {
|
|
/* First we have to store the value of the loop index in a new variable
|
|
* which will be private for each task because otherwise it will be overwritten
|
|
* if the execution of the task takes longer than the time which is needed to
|
|
* enter the next step of the loop!
|
|
*/
|
|
int myi;
|
|
myi = i;
|
|
#pragma omp task
|
|
{
|
|
my_sleep (SLEEPTIME);
|
|
array[myi] = 1;
|
|
} /* end of omp task */
|
|
} /* end of for */
|
|
#pragma omp taskwait
|
|
/* check if all tasks were finished */
|
|
for (i = 0; i < NUM_TASKS; i++)
|
|
if (array[i] != 1)
|
|
result1++;
|
|
|
|
/* generate some more tasks which now shall overwrite
|
|
* the values in the tids array */
|
|
for (i = 0; i < NUM_TASKS; i++) {
|
|
int myi;
|
|
myi = i;
|
|
#pragma omp task
|
|
{
|
|
array[myi] = 2;
|
|
} /* end of omp task */
|
|
} /* end of for */
|
|
} /* end of single */
|
|
} /*end of parallel */
|
|
|
|
/* final check, if all array elements contain the right values: */
|
|
for (i = 0; i < NUM_TASKS; i++) {
|
|
if (array[i] != 2)
|
|
result2++;
|
|
}
|
|
return ((result1 == 0) && (result2 == 0));
|
|
}
|
|
|
|
int main()
|
|
{
|
|
int i;
|
|
int num_failed=0;
|
|
|
|
for(i = 0; i < REPETITIONS; i++) {
|
|
if(!test_omp_taskwait()) {
|
|
num_failed++;
|
|
}
|
|
}
|
|
return num_failed;
|
|
}
|