1  /* { dg-do run } */
       2  
       3  /* This tests usage of ICVs on the host and on devices if no corresponding
       4     environment variables are configured.  */
       5  
       6  #include <omp.h>
       7  #include <stdlib.h>
       8  
       9  int
      10  main ()
      11  {
      12    if (omp_get_max_teams () != 0
      13        || omp_get_teams_thread_limit () != 0)
      14      abort ();
      15  
      16    omp_set_num_teams (9);
      17    omp_set_teams_thread_limit (2);
      18    if (omp_get_max_teams () != 9
      19        || omp_get_teams_thread_limit () != 2)
      20      abort ();
      21  
      22    #pragma omp teams
      23    if (omp_get_num_teams () > 9
      24        || omp_get_team_num () >= 9)
      25      abort ();
      26  
      27    #pragma omp teams num_teams(5)
      28    if (omp_get_num_teams () > 5
      29        || omp_get_team_num () >= 5)
      30      abort ();
      31  
      32    int num_devices = omp_get_num_devices () > 3 ? 3 : omp_get_num_devices ();
      33    for (int i = 0; i < num_devices; i++)
      34      {
      35        #pragma omp target device (i)
      36        if (omp_get_max_teams () != 0
      37  	  || omp_get_teams_thread_limit () != 0)
      38  	abort ();
      39  
      40        #pragma omp target device (i)
      41        {
      42  	omp_set_num_teams (8 + i);
      43  	omp_set_teams_thread_limit (3 + i);
      44  	if (omp_get_max_teams () != 8 + i
      45  	    || omp_get_teams_thread_limit () != 3 + i)
      46  	  abort ();
      47        }
      48  
      49       /* omp_set_num_teams above set the value of nteams-var ICV on device 'i',
      50  	 which has scope 'device' and should be avaible in subsequent target
      51  	 regions.  */
      52        #pragma omp target device (i)
      53        if (omp_get_max_teams () != 8 + i
      54  	  || omp_get_teams_thread_limit () != 3 + i)
      55  	abort ();
      56  
      57        #pragma omp target device (i)
      58        #pragma omp teams
      59        if (omp_get_num_teams () > 8 + i
      60  	  || omp_get_team_num () >= 8 + i)
      61  	abort ();
      62  
      63        /* NUM_TEAMS clause has priority over previously set NUM_TEAMS value.  */
      64        #pragma omp target device (i)
      65        #pragma omp teams num_teams(5 + i)
      66        if (omp_get_num_teams () > 5 + i
      67  	  || omp_get_team_num () >= 5 + i)
      68  	abort ();
      69      }
      70  
      71    return 0;
      72  }