OpenMP-Examples/sources/Example_metadirective.2.c

34 lines
936 B
C

/*
* @@name: metadirective.2c
* @@type: C
* @@compilable: yes, omp_5.0
* @@linkable: no
* @@expect: success
*/
#define N 100
#include <stdio.h>
#include <omp.h>
void work_on_chunk(int idev, int i);
int main() //Driver
{
int i,idev;
for (idev=0; idev<omp_get_num_devices()-1; idev++)
{
#pragma omp target device(i)
#pragma omp metadirective \
when( implementation={vendor(nvidia)}, device={arch("kepler")}: \
teams num_teams(512) thread_limit(32) ) \
when( implementation={vendor(amd)}, device={arch("fiji" )}: \
teams num_teams(512) thread_limit(64) ) \
default( \
teams)
#pragma omp distribute parallel for
for (i=0; i<N; i++) work_on_chunk(idev,i);
}
}