1  /* { dg-do run { target openacc_nvidia_accel_selected } } */
       2  /* { dg-additional-options "-lcuda" } */
       3  /* { dg-require-effective-target openacc_cuda } */
       4  
       5  #include <stdlib.h>
       6  #include <unistd.h>
       7  #include <openacc.h>
       8  #include <stdio.h>
       9  #include <cuda.h>
      10  
      11  #if !defined __cplusplus
      12  # undef static_assert
      13  # define static_assert _Static_assert
      14  #endif
      15  
      16  static_assert (acc_async_sync == -2, "acc_async_sync?");
      17  static_assert (acc_async_noval == -1, "acc_async_noval?");
      18  
      19  int
      20  main (int argc, char **argv)
      21  {
      22    const int N = 100;
      23    int i;
      24    CUstream *streams;
      25    CUstream s;
      26    CUresult r;
      27  
      28    acc_init (acc_device_nvidia);
      29  
      30    (void) acc_get_device_num (acc_device_nvidia);
      31  
      32    streams = (CUstream *) malloc ((2 + N) * sizeof (void *));
      33    streams += 2;
      34    /* "streams[i]" is valid for i in [acc_async_sync..N).  */
      35  
      36    for (i = acc_async_sync; i < N; i++)
      37      {
      38        streams[i] = (CUstream) acc_get_cuda_stream (i);
      39        if (streams[i] != NULL)
      40  	abort ();
      41  
      42        r = cuStreamCreate (&streams[i], CU_STREAM_DEFAULT);
      43        if (r != CUDA_SUCCESS)
      44  	{
      45  	  fprintf (stderr, "cuStreamCreate failed: %d\n", r);
      46  	  abort ();
      47  	}
      48  
      49        int ret = acc_set_cuda_stream (i, streams[i]);
      50        if (i == acc_async_sync)
      51  	{
      52  	  if (ret == 1)
      53  	    abort ();
      54  	}
      55        else
      56  	{
      57  	  if (ret != 1)
      58  	    abort ();
      59  	}
      60      }
      61  
      62    s = NULL;
      63  
      64    if (acc_set_cuda_stream (N + 1, s) != 0)
      65      abort ();
      66  
      67    acc_shutdown (acc_device_nvidia);
      68  
      69    exit (0);
      70  }
      71  
      72  /* { dg-output "" } */