diff --git a/.gitmodules b/.gitmodules index f511007e5278..3bf2c4ef6465 100644 --- a/.gitmodules +++ b/.gitmodules @@ -32,3 +32,6 @@ [submodule "third_party/libxsmm"] path = third_party/libxsmm url = https://github.com/hfp/libxsmm.git +[submodule "third_party/thrust"] + path = third_party/thrust + url = https://github.com/NVIDIA/thrust.git diff --git a/CMakeLists.txt b/CMakeLists.txt index 74c33e6defec..af0998deb7ff 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -54,6 +54,7 @@ if(USE_CUDA) # is fixed by https://github.com/NVIDIA/cub/commit/9143e47e048641aa0e6ddfd645bcd54ff1059939 # in 11.1. message(STATUS "Detected CUDA of version ${CUDA_VERSION}. Use external CUB/Thrust library.") + cuda_include_directories(BEFORE "${CMAKE_SOURCE_DIR}/third_party/thrust") cuda_include_directories(BEFORE "${CMAKE_SOURCE_DIR}/third_party/cub") endif() endif(USE_CUDA) diff --git a/third_party/thrust b/third_party/thrust new file mode 160000 index 000000000000..0ef5c509856e --- /dev/null +++ b/third_party/thrust @@ -0,0 +1 @@ +Subproject commit 0ef5c509856e12cc408f0f00ed586b4c5b1a155c