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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
/* { dg-additional-options "-fdump-tree-gimple" } */
#if __cplusplus >= 201103L
# define __GOMP_UINTPTR_T_ENUM : __UINTPTR_TYPE__
#else
# define __GOMP_UINTPTR_T_ENUM
#endif
typedef enum omp_interop_t __GOMP_UINTPTR_T_ENUM
{
omp_interop_none = 0,
__omp_interop_t_max__ = __UINTPTR_MAX__
} omp_interop_t;
typedef enum omp_interop_fr_t
{
omp_ifr_cuda = 1,
omp_ifr_cuda_driver = 2,
omp_ifr_opencl = 3,
omp_ifr_sycl = 4,
omp_ifr_hip = 5,
omp_ifr_level_zero = 6,
omp_ifr_hsa = 7,
omp_ifr_last = omp_ifr_hsa
} omp_interop_fr_t;
extern int flag;
void g1(int, const char *, int *, int *, omp_interop_t, omp_interop_t) { }
void g2(int, const char *, int *, int *, omp_interop_t, omp_interop_t) { }
#pragma omp declare variant(g1) \
match(construct={dispatch}, user={condition(flag==1)}) \
append_args(interop(target,prefer_type( {fr("cuda") }, {fr(omp_ifr_hsa)} , {attr("ompx_a") } , {fr(omp_ifr_hip) }), targetsync), \
interop(targetsync, prefer_type("cuda", "hsa"))) \
adjust_args(need_device_ptr : y, k)
#pragma omp declare variant(g2) \
match(construct={dispatch}, user={condition(flag==2)}) \
append_args(interop(target,prefer_type( {fr("cuda") }, {fr(omp_ifr_hsa)} , {attr("ompx_a") } , {fr(omp_ifr_hip) }), targetsync), \
interop(targetsync, prefer_type("cuda", "hsa"))) \
adjust_args(need_device_ptr : y)
void f(int x, const char *y, int *, int *k) { }
void gvar1(int, const char *, int *, int *, omp_interop_t, omp_interop_t, ...) { }
void gvar2(int, const char *, int *, int *, omp_interop_t, omp_interop_t, ...) { }
#pragma omp declare variant(gvar1) \
match(construct={dispatch}, user={condition(flag==3)}) \
append_args(interop(target,prefer_type( {fr("cuda") }, {fr(omp_ifr_hsa)} , {attr("ompx_a") } , {fr(omp_ifr_hip) }), targetsync), \
interop(targetsync, prefer_type("cuda", "hsa"))) \
adjust_args(need_device_ptr : y, k)
#pragma omp declare variant(gvar2) \
match(construct={dispatch}, user={condition(flag==4)}) \
append_args(interop(target,prefer_type( {fr("cuda") }, {fr(omp_ifr_hsa)} , {attr("ompx_a") } , {fr(omp_ifr_hip) }), targetsync), \
interop(targetsync, prefer_type("cuda", "hsa"))) \
adjust_args(need_device_ptr : y)
void fvar(int x, const char *y, int *, int *k, ...) { }
void foo(const char *cp1, const char *cp2, int *a, int *b, int *c)
{
omp_interop_t obj1, obj2, obj3, obj4;
obj1 = obj2 = obj3 = obj4 = omp_interop_none;
#pragma omp dispatch device(5) interop(obj1,obj2) is_device_ptr(cp1)
f(3, cp1, a, b);
#pragma omp dispatch device(4) interop(obj3,obj4) is_device_ptr(a,b,c)
fvar(99, cp2, a, b, c, a, b, c, a, b, c);
}
/* Since the selectors are dynamic, there must be calls to all the variants as
well as the base functions. */
/* { dg-final { scan-tree-dump "f \\(3, cp1, a, b\\)" "gimple" } } */
/* { dg-final { scan-tree-dump "g1 \\(3, cp1, a, D\.\[0-9\]+, obj1, obj2\\)" "gimple" } } */
/* { dg-final { scan-tree-dump "g2 \\(3, cp1, a, b, obj1, obj2\\)" "gimple" } } */
/* { dg-final { scan-tree-dump "fvar \\(99, cp2, a, b, c, a, b, c, a, b, c\\)" "gimple" } } */
/* { dg-final { scan-tree-dump "gvar1 \\(99, D\.\[0-9\]+, a, b, obj3, obj4, c, a, b, c, a, b, c\\)" "gimple" } } */
/* { dg-final { scan-tree-dump "gvar2 \\(99, D\.\[0-9\]+, a, b, obj3, obj4, c, a, b, c, a, b, c\\)" "gimple" } } */
/* Check that the condition tests appear in the output, too. */
/* { dg-final { scan-tree-dump "if \\(flag.\[0-9\]+ == 1\\)" "gimple" } } */
/* { dg-final { scan-tree-dump "if \\(flag.\[0-9\]+ == 2\\)" "gimple" } } */
/* { dg-final { scan-tree-dump "if \\(flag.\[0-9\]+ == 3\\)" "gimple" } } */
/* { dg-final { scan-tree-dump "if \\(flag.\[0-9\]+ == 4\\)" "gimple" } } */
/* There should be one call to map the pointer b for the call to f's variant
g1; g2 doesn't have a need_device_ptr on this parameter. */
/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_mapped_ptr \\(b," 1 "gimple" } } */
/* Both variants for fvar need to map pointer cp2. */
/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_mapped_ptr \\(cp2," 2 "gimple" } } */
|
