diff options
Diffstat (limited to 'nerv/lib/matrix/cuda_helper.h')
| -rw-r--r-- | nerv/lib/matrix/cuda_helper.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/nerv/lib/matrix/cuda_helper.h b/nerv/lib/matrix/cuda_helper.h index 8041efb..13d5728 100644 --- a/nerv/lib/matrix/cuda_helper.h +++ b/nerv/lib/matrix/cuda_helper.h @@ -4,6 +4,7 @@ #include "cuda_runtime.h" #include "driver_types.h" #include "cublas_v2.h" +#include "curand.h" #define CUBLAS_SAFE_SYNC_CALL_RET(call, status) \ do { \ |