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