libgomp/testsuite/libgomp.oacc-c-c++-common/lib-90.c

   1 /* { dg-do run { target openacc_nvidia_accel_selected } } */
   2 /* { dg-additional-options "-lcuda" } */
   3
   4 #include <pthread.h>
   5 #include <stdio.h>
   6 #include <string.h>
   7 #include <stdlib.h>
   8 #include <unistd.h>
   9 #include <errno.h>
  10 #include <ctype.h>
  11 #include <openacc.h>
  12 #include <cuda.h>
  13
  14 unsigned char **x;
  15 void **d_x;
  16 const int N = 16;
  17 const int NTHREADS = 32;
  18
  19 static void *
  20 test (void *arg)
  21 {
  22   int i;
  23   int tid;
  24   unsigned char *p;
  25   int devnum;
  26
  27   tid = (int) (long) arg;
  28
  29   devnum = acc_get_device_num (acc_device_nvidia);
  30   acc_set_device_num (devnum, acc_device_nvidia);
  31
  32   if (acc_get_current_cuda_context () == NULL)
  33     abort ();
  34
  35   p = (unsigned char *) malloc (N);
  36
  37   for (i = 0; i < N; i++)
  38     {
  39       p[i] = tid;
  40     }
  41
  42   x[tid] = p;
  43
  44   d_x[tid] = acc_copyin (p, N);
  45
  46   acc_wait_all ();
  47
  48   return 0;
  49 }
  50
  51 int
  52 main (int argc, char **argv)
  53 {
  54   int i;
  55   pthread_attr_t attr;
  56   pthread_t *tid;
  57   CUresult r;
  58   CUstream s;
  59
  60   acc_init (acc_device_nvidia);
  61
  62   x = (unsigned char **) malloc (NTHREADS * N);
  63   d_x = (void **) malloc (NTHREADS * N);
  64
  65   if (pthread_attr_init (&attr) != 0)
  66     perror ("pthread_attr_init failed");
  67
  68   tid = (pthread_t *) malloc (NTHREADS * sizeof (pthread_t));
  69
  70   r = cuStreamCreate (&s, CU_STREAM_DEFAULT);
  71   if (r != CUDA_SUCCESS)
  72         {
  73           fprintf (stderr, "cuStreamCreate failed: %d\n", r);
  74           abort ();
  75         }
  76
  77   if (!acc_set_cuda_stream (0, s))
  78           abort ();
  79
  80   for (i = 0; i < NTHREADS; i++)
  81     {
  82       if (pthread_create (&tid[i], &attr, &test, (void *) (unsigned long) (i))
  83           != 0)
  84         perror ("pthread_create failed");
  85     }
  86
  87   if (pthread_attr_destroy (&attr) != 0)
  88     perror ("pthread_attr_destroy failed");
  89
  90   for (i = 0; i < NTHREADS; i++)
  91     {
  92       void *res;
  93
  94       if (pthread_join (tid[i], &res) != 0)
  95         perror ("pthread join failed");
  96     }
  97
  98
  99   for (i = 0; i < NTHREADS; i++)
 100     {
 101       if (acc_is_present (x[i], N) != 1)
 102         abort ();
 103     }
 104
 105   acc_get_cuda_stream (1);
 106
 107   for (i = 0; i < NTHREADS; i++)
 108     {
 109       memset (x[i], 0, N);
 110       acc_copyout (x[i], N);
 111     }
 112
 113   acc_wait_all ();
 114
 115   for (i = 0; i < NTHREADS; i++)
 116     {
 117       unsigned char *p;
 118       int j;
 119
 120       p = x[i];
 121
 122       for (j = 0; j < N; j++)
 123         {
 124           if (p[j] != i)
 125             abort ();
 126         }
 127
 128       if (acc_is_present (x[i], N) != 0)
 129         abort ();
 130     }
 131
 132   acc_shutdown (acc_device_nvidia);
 133
 134   return 0;
 135 }
 136
 137 /* { dg-output "" } */