diff options
| author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2016-05-04 20:21:24 +0300 |
|---|---|---|
| committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2016-05-04 20:21:24 +0300 |
| commit | 4291c918aefefd52da1c4334743e0a2960a9abcc (patch) | |
| tree | caf5e308e601f56b59f770336953a0930fc8e01c /src | |
| parent | a43d0b27c04ff6c753449dbde8c27c324893069c (diff) | |
towards cudnn
Diffstat (limited to 'src')
| -rw-r--r-- | src/test.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/test.cpp b/src/test.cpp index 795d845d..8d1e380c 100644 --- a/src/test.cpp +++ b/src/test.cpp @@ -1,6 +1,12 @@ #include <iostream> #include <ctime> +#include <cuda_runtime.h> +#include <device_launch_parameters.h> + +#include <cublas_v2.h> +#include <cudnn.h> + #include "marian.h" using namespace marian; |