[Build] Fix NCCL building crashes when using submodules but with system NCCL installed (#2975)

7e58236c · Quan (Andy) Gan · GitHub · 75ec5826 · 7e58236c
Unverified Commit 7e58236c authored Jun 03, 2021 by Quan (Andy) Gan Committed by GitHub Jun 03, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

CMakeLists.txt CMakeLists.txt +2 -1

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -162,11 +162,12 @@ if(USE_CUDA)
  if (USE_SYSTEM_NCCL)
    include(cmake/util/FindNccl.cmake)
+    include_directories(${NCCL_INCLUDE_DIR})
  else()
    include(cmake/modules/NCCL.cmake)
+    cuda_include_directories(BEFORE ${NCCL_INCLUDE_DIR})
  endif()
-  include_directories(${NCCL_INCLUDE_DIR})
  list(APPEND DGL_LINKER_LIBS ${NCCL_LIBRARY})
 endif(USE_CUDA)