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>