Ver Fonte

CUDAToolkit: Restore ability to find cupti headers

Fixes #25484

PR !8835 failed to update the CUPTI header searches to use the
new internal FindCUDAToolkit search variables. This caused the
CUDA::cupti target to always not exist.
Robert Maynard há 1 ano atrás
pai
commit
f69b9b7305
1 ficheiros alterados com 3 adições e 2 exclusões
  1. 3 2
      Modules/FindCUDAToolkit.cmake

+ 3 - 2
Modules/FindCUDAToolkit.cmake

@@ -1244,8 +1244,9 @@ if(CUDAToolkit_FOUND)
 
   find_path(CUDAToolkit_CUPTI_INCLUDE_DIR cupti.h PATHS
       "${CUDAToolkit_ROOT_DIR}/extras/CUPTI/include"
-      "${CUDAToolkit_INCLUDE_DIR}/../extras/CUPTI/include"
-      "${CUDAToolkit_INCLUDE_DIR}"
+      "${CUDAToolkit_INCLUDE_DIRS}"
+      PATH_SUFFIXES "../extras/CUPTI/include"
+                    "../../../extras/CUPTI/include"
       NO_DEFAULT_PATH)
   mark_as_advanced(CUDAToolkit_CUPTI_INCLUDE_DIR)