From c8365d2c5b2568eb96a4ec11894cef97ec425a12 Mon Sep 17 00:00:00 2001
From: KazukiYoshiyama <Kazuki.Yoshiyama@jp.sony.com>
Date: Fri, 26 Oct 2018 14:29:19 +0900
Subject: [PATCH] Add include.

---
 src/nbla/cuda/function/generic/sort.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/nbla/cuda/function/generic/sort.cu b/src/nbla/cuda/function/generic/sort.cu
index 570562ba3..97700838a 100644
--- a/src/nbla/cuda/function/generic/sort.cu
+++ b/src/nbla/cuda/function/generic/sort.cu
@@ -17,6 +17,8 @@
 #include <nbla/cuda/function/sort.hpp>
 #include <nbla/variable.hpp>
 
+#include <thrust/device_ptr.h>
+#include <thrust/device_vector.h>
 #include <thrust/sequence.h>
 #include <thrust/sort.h>