forked from OSchip/llvm-project
62 lines
1.3 KiB
C
62 lines
1.3 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_taskyield()
|
|
{
|
|
int i;
|
|
int count = 0;
|
|
int start_tid[NUM_TASKS];
|
|
int current_tid[NUM_TASKS];
|
|
|
|
for (i=0; i< NUM_TASKS; i++) {
|
|
start_tid[i]=0;
|
|
current_tid[i]=0;
|
|
}
|
|
|
|
#pragma omp parallel
|
|
{
|
|
#pragma omp single
|
|
{
|
|
for (i = 0; i < NUM_TASKS; i++) {
|
|
int myi = i;
|
|
#pragma omp task untied
|
|
{
|
|
my_sleep(SLEEPTIME);
|
|
start_tid[myi] = omp_get_thread_num();
|
|
#pragma omp taskyield
|
|
if((start_tid[myi] %2) ==0){
|
|
my_sleep(SLEEPTIME);
|
|
current_tid[myi] = omp_get_thread_num();
|
|
} /*end of if*/
|
|
} /* end of omp task */
|
|
} /* end of for */
|
|
} /* end of single */
|
|
} /* end of parallel */
|
|
for (i=0;i<NUM_TASKS; i++) {
|
|
//printf("start_tid[%d]=%d, current_tid[%d]=%d\n",
|
|
//i, start_tid[i], i , current_tid[i]);
|
|
if (current_tid[i] == start_tid[i])
|
|
count++;
|
|
}
|
|
return (count<NUM_TASKS);
|
|
}
|
|
|
|
int main()
|
|
{
|
|
int i;
|
|
int num_failed=0;
|
|
|
|
if (omp_get_max_threads() < 2)
|
|
omp_set_num_threads(8);
|
|
|
|
for(i = 0; i < REPETITIONS; i++) {
|
|
if(!test_omp_taskyield()) {
|
|
num_failed++;
|
|
}
|
|
}
|
|
return num_failed;
|
|
}
|