AlexeyAB
2018-04-19 5d616450a43f4b5fdf130f57cdaab105cf2563e2
src/cuda.c
@@ -96,6 +96,7 @@
    int i = cuda_get_device();
    if(!init[i]) {
        cublasCreate(&handle[i]);
      cublasStatus_t status = cublasSetStream(handle[i], get_cuda_stream());
        init[i] = 1;
    }
    return handle[i];
@@ -176,4 +177,7 @@
   cudaStreamSynchronize(get_cuda_stream());
}
#endif
#else // GPU
#include "cuda.h"
void cuda_set_device(int n) {}
#endif // GPU