AlexeyAB
2018-06-15 e1e5abe19197f2bb56a9d320b31275db4706aa4d
src/cuda.c
@@ -67,7 +67,16 @@
cudaStream_t get_cuda_stream() {
   int i = cuda_get_device();
   if (!streamInit[i]) {
      cudaStreamCreate(&streamsArray[i]);
      cudaError_t status = cudaStreamCreate(&streamsArray[i]);
      //cudaError_t status = cudaStreamCreateWithFlags(&streamsArray[i], cudaStreamNonBlocking);
      if (status != cudaSuccess) {
         printf(" cudaStreamCreate error: %d \n", status);
         const char *s = cudaGetErrorString(status);
         char buffer[256];
         printf("CUDA Error: %s\n", s);
         status = cudaStreamCreateWithFlags(&streamsArray[i], cudaStreamDefault);
         check_error(status);
      }
      streamInit[i] = 1;
   }
   return streamsArray[i];