Skip to content

Commit bf4050e

Browse files
samuelapytorchmergebot
authored andcommitted
Dependencies.cmake: support building against CUPTI outside of CUDA_SOURCE_DIR
Limitation discovered in NixOS/nixpkgs#249259.
1 parent 1eb6c43 commit bf4050e

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

cmake/Dependencies.cmake

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1879,13 +1879,15 @@ if(USE_KINETO)
18791879
${CUDA_SOURCE_DIR}/extras/CUPTI/lib64
18801880
${CUDA_SOURCE_DIR}/lib
18811881
${CUDA_SOURCE_DIR}/lib64
1882+
$ENV{CUPTI_LIBRARY_DIR}
18821883
NO_DEFAULT_PATH)
18831884

18841885
find_path(CUPTI_INCLUDE_DIR cupti.h PATHS
18851886
${CUDA_SOURCE_DIR}/extras/CUPTI/include
18861887
${CUDA_INCLUDE_DIRS}
18871888
${CUDA_SOURCE_DIR}
18881889
${CUDA_SOURCE_DIR}/include
1890+
$ENV{CUPTI_INCLUDE_DIR}
18891891
NO_DEFAULT_PATH)
18901892

18911893
if(CUPTI_LIBRARY_PATH AND CUPTI_INCLUDE_DIR)

0 commit comments

Comments
 (0)