gcc/testsuite/c-c++-common/gomp/append-args-6.c

/* { 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;



void g(int, const char *, int *, int *, omp_interop_t, omp_interop_t) { }
#pragma omp declare variant(g) match(construct={dispatch}) \
   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)
void f(int x, const char *y, int *, int *k) { }


void gvar(int, const char *, int *, int *, omp_interop_t, omp_interop_t, ...) { }
#pragma omp declare variant(gvar) match(construct={dispatch}) \
   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)
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);
}




int *fi();

struct t {
  int *a, *b;
};

void fancy(int *x, int *y, omp_interop_t)  { }

#pragma omp declare variant(fancy) match(construct={dispatch}) adjust_args(need_device_ptr: x,y) \
                    append_args( interop (prefer_type(omp_ifr_hip), target) )
void bar(int *x, int *y);

void sub(struct t *s, void *y, const omp_interop_t obj5, omp_interop_t obj6)
{
 bar( fi(), s->b);

 // This is a bit questionable as dereferencing 's' as device pointer might not work (unspecified behavior);
 // but if for 's->b' it would still be need even if 's' was a device + host accessible pointer.
 #pragma omp dispatch device(3) is_device_ptr(s) interop(obj5)
    bar( fi(), s->b);

 bar( (int *) y, s->b);
 #pragma omp dispatch interop(obj6) is_device_ptr(y)
    bar( (int *) y, s->b);
}


/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_default_device \\(\\);" 4 "gimple" } }  */

/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_interop_int \\(obj6, -5, 0B\\);" 1 "gimple" } }  */

/* { dg-final { scan-tree-dump-times "__builtin_omp_set_default_device \\(5\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "__builtin_omp_set_default_device \\(4\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "__builtin_omp_set_default_device \\(3\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "__builtin_omp_set_default_device \\(D\.\[0-9\]+\\);" 5 "gimple" } }  */

/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_mapped_ptr \\(b, 5\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_mapped_ptr \\(cp2, 4\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_mapped_ptr \\(cp2, 4\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_mapped_ptr \\(cp2, 4\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_mapped_ptr \\(D\.\[0-9\]+, 3\\);" 2 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "D\.\[0-9\]+ = __builtin_omp_get_mapped_ptr \\(D\.\[0-9\]+, D\.\[0-9\]+\\);" 1 "gimple" } }  */

/* { dg-final { scan-tree-dump-times "g \\(3, cp1, a, D\.\[0-9\]+, obj1, obj2\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "gvar \\(99, D\.\[0-9\]+, a, b, obj3, obj4, c, a, b, c, a, b, c\\);" 1 "gimple" } }  */

/* { dg-final { scan-tree-dump-times "fancy \\(D\.\[0-9\]+, D\.\[0-9\]+, obj5\\);" 1 "gimple" } }  */
/* { dg-final { scan-tree-dump-times "fancy \\(y, D\.\[0-9\]+, obj6\\);" 1 "gimple" } }  */