Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "18ebd57bd80fdd1bb8cbf9af075ba0301705bc6f"
Unverified Commit 769718df authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

Bring back thrust for backward compatibility (#3562)

parent 9f445a13
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
.gitmodules
View file @ 769718df
...@@ -32,3 +32,6 @@ ...@@ -32,3 +32,6 @@
[submodule "third_party/libxsmm"] [submodule "third_party/libxsmm"]
path = third_party/libxsmm path = third_party/libxsmm
url = https://github.com/hfp/libxsmm.git url = https://github.com/hfp/libxsmm.git
[submodule "third_party/thrust"]
path = third_party/thrust
url = https://github.com/NVIDIA/thrust.git
CMakeLists.txt
View file @ 769718df
...@@ -54,6 +54,7 @@ if(USE_CUDA) ...@@ -54,6 +54,7 @@ if(USE_CUDA)
# is fixed by https://github.com/NVIDIA/cub/commit/9143e47e048641aa0e6ddfd645bcd54ff1059939 # is fixed by https://github.com/NVIDIA/cub/commit/9143e47e048641aa0e6ddfd645bcd54ff1059939
# in 11.1. # in 11.1.
message(STATUS "Detected CUDA of version ${CUDA_VERSION}. Use external CUB/Thrust library.") 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") cuda_include_directories(BEFORE "${CMAKE_SOURCE_DIR}/third_party/cub")
endif() endif()
endif(USE_CUDA) endif(USE_CUDA)
......
thrust @ 0ef5c509
Subproject commit 0ef5c509856e12cc408f0f00ed586b4c5b1a155c
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment