Skip to content

Commit

Permalink
Add cuCollections dependency (#371)
Browse files Browse the repository at this point in the history
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
  • Loading branch information
rusty1s and pre-commit-ci[bot] authored Dec 30, 2024
1 parent fdec9bd commit 7c34d84
Show file tree
Hide file tree
Showing 5 changed files with 22 additions and 0 deletions.
6 changes: 6 additions & 0 deletions .gitmodules
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,9 @@
[submodule "third_party/METIS"]
path = third_party/METIS
url = https://github.com/KarypisLab/METIS.git
[submodule "third_party/cuCollections"]
path = third_party/cuCollections
url = https://github.com/NVIDIA/cuCollections.git
[submodule "third_party/cccl"]
path = third_party/cccl
url = https://github.com/NVIDIA/cccl.git
9 changes: 9 additions & 0 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,8 @@ if(WITH_CUDA)
set(CUTLASS_UTIL_DIR third_party/cutlass/tools/util/include)
include_directories(${CUTLASS_UTIL_DIR})
endif()
set(CUCOLLECTIONS_DIR third_party/cuCollections/include)
include_directories(${CUCOLLECTIONS_DIR})
endif()

set(CSRC pyg_lib/csrc)
Expand Down Expand Up @@ -105,6 +107,13 @@ if (USE_PYTHON)
target_link_libraries(${PROJECT_NAME} PRIVATE Python3::Python)
endif()

if(WITH_CUDA)
target_include_directories(${PROJECT_NAME} PRIVATE
third_party/cccl/thrust
third_party/cccl/cub
third_party/cccl/libcudacxx/include)
endif()

set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -O0")

if(BUILD_TEST)
Expand Down
5 changes: 5 additions & 0 deletions MANIFEST.in
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,11 @@ recursive-exclude third_party/parallel-hashmap/tests *
recursive-exclude third_party/parallel-hashmap/examples *
recursive-exclude third_party/parallel-hashmap/benchmark *
recursive-exclude third_party/METIS/graphs *
recursive-exclude third_party/cccl/benchmarks *
recursive-exclude third_party/cccl/docs *
recursive-exclude third_party/cccl/examples *
recursive-exclude third_party/cuCollections/examples *
recursive-exclude third_party/cuCollections/tests *

recursive-exclude test *
recursive-exclude docs *
Expand Down
1 change: 1 addition & 0 deletions third_party/cccl
Submodule cccl added at faca86
1 change: 1 addition & 0 deletions third_party/cuCollections
Submodule cuCollections added at e79787

0 comments on commit 7c34d84

Please sign in to comment.