add missing include for rocthrust

apache · Feb 17, 2021 · 27fb5b1 · 27fb5b1
1 parent 684ecc3
commit 27fb5b1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/runtime/contrib/thrust/thrust.cu b/src/runtime/contrib/thrust/thrust.cu
@@ -26,6 +26,7 @@
 #include <thrust/sort.h>
 #include <thrust/gather.h>
 #include <thrust/scan.h>
+#include <thrust/sequence.h>
 
 #include <tvm/runtime/registry.h>
 #include <dlpack/dlpack.h>