50 lines · c
1// RUN: %libomp-compile && env OMP_DYNAMIC=true KMP_DYNAMIC_MODE=random %libomp-run2 3// gcc/icc target offloading is incompatible with libomp4// UNSUPPORTED: icc, gcc5 6// This is a super simple unit test to see that teams behave properly when7// parallel regions inside the teams construct cannot allocate teams of8// thread_limit size.9 10#include <stdio.h>11#include <stdlib.h>12#include <omp.h>13 14#define NUM_TIMES 1015 16int main(int argc, char **argv) {17 int num_procs = omp_get_max_threads();18 int num_teams, thread_limit, i;19 num_teams = 2;20 thread_limit = num_procs / num_teams;21 for (i = 0; i < NUM_TIMES; ++i) {22#pragma omp target teams num_teams(num_teams) thread_limit(thread_limit)23 {24#pragma omp parallel num_threads(thread_limit)25 {26 int my_num_threads = omp_get_num_threads();27 int my_num_teams = omp_get_num_teams();28 int my_team_id = omp_get_team_num();29 int my_thread_id = omp_get_thread_num();30 if (my_num_teams < 0 || my_num_teams > num_teams) {31 fprintf(stderr, "error: my_num_teams (%d) invalid\n", my_num_teams);32 exit(1);33 }34 if (my_team_id < 0 || my_team_id >= my_num_teams) {35 fprintf(stderr, "error: my_team_id (%d) invalid (nteams = %d)\n",36 my_team_id, my_num_teams);37 exit(1);38 }39 if (my_thread_id < 0 || my_thread_id >= my_num_threads) {40 fprintf(stderr,41 "error: my_thread_id (%d) invalid (my_num_threads = %d)\n",42 my_thread_id, my_num_threads);43 exit(1);44 }45 }46 }47 }48 return 0;49}50