blob: 73c105d58741be3fed54917227d8461b06d95773 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
#include <omp.h>
#include <stdlib.h>
int main (void)
{
int host_device_num = omp_get_device_num ();
if (host_device_num != omp_get_initial_device ())
abort ();
int device_num;
int initial_device;
for (int i = 0; i <= omp_get_num_devices (); i++)
{
#pragma omp target map(from: device_num, initial_device) device(i)
{
initial_device = omp_is_initial_device ();
device_num = omp_get_device_num ();
}
if (i != device_num)
abort ();
if (initial_device && host_device_num != device_num)
abort ();
if (!initial_device && host_device_num == device_num)
abort ();
}
return 0;
}
|