From 4291c918aefefd52da1c4334743e0a2960a9abcc Mon Sep 17 00:00:00 2001
From: Marcin Junczys-Dowmunt <junczys@amu.edu.pl>
Date: Wed, 4 May 2016 18:21:24 +0100
Subject: [PATCH] towards cudnn

---
 src/test.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

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;