add missing include for rocthrust

Author: Masahiro Masuda

cmake/modules/ROCM.cmake

@@ -61,7 +61,7 @@ if(USE_ROCM)
     find_package(rocprim REQUIRED)
     find_package(rocthrust REQUIRED)
     set_source_files_properties(src/runtime/contrib/thrust/thrust.cu PROPERTIES LANGUAGE CXX)
-    file(GLOB CONTRIB_THRUST_SRC src/runtime/contrib/thrust/thrust.cu)
+    file(GLOB CONTRIB_THRUST_SRC src/runtime/contrib/thrust/*.cu)
     list(APPEND RUNTIME_SRCS ${CONTRIB_THRUST_SRC})
     list(APPEND TVM_RUNTIME_LINKER_LIBS roc::rocthrust)
   endif(USE_THRUST)

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>