浏览代码

FindCUDAToolkit: Allow new search locations after failure

Fixes #22010
Robert Maynard 1 年之前
父节点
当前提交
8f974011ea
共有 1 个文件被更改,包括 8 次插入0 次删除
  1. 8 0
      Modules/FindCUDAToolkit.cmake

+ 8 - 0
Modules/FindCUDAToolkit.cmake

@@ -1101,6 +1101,14 @@ if(CUDAToolkit_FOUND)
     unset(CUDAToolkit_search_loc)
     unset(CUDAToolkit_possible_lib_root)
   endif()
+else()
+  # clear cache results when we fail
+  unset(_cmake_CUDAToolkit_implicit_link_directories CACHE)
+  unset(_cmake_CUDAToolkit_include_directories CACHE)
+  unset(CUDA_CUDART CACHE)
+  unset(CUDAToolkit_BIN_DIR CACHE)
+  unset(CUDAToolkit_NVCC_EXECUTABLE CACHE)
+  unset(CUDAToolkit_SENTINEL_FILE CACHE)
 endif()
 unset(CUDAToolkit_IMPLICIT_LIBRARY_DIRECTORIES)
 unset(CUDAToolkit_INCLUDE_DIRECTORIES)