Abort when nvcc command is not found in the PATH (#3527)

84eaa684 · Allen.Dou · GitHub · 5ee14494 · 84eaa684
Unverified Commit 84eaa684 authored Mar 21, 2024 by Allen.Dou Committed by GitHub Mar 20, 2024
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

CMakeLists.txt CMakeLists.txt +6 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -49,6 +49,12 @@ endif()
 #
 append_cmake_prefix_path("torch" "torch.utils.cmake_prefix_path")

+# Ensure the 'nvcc' command is in the PATH
+find_program(NVCC_EXECUTABLE nvcc)
+if (NOT NVCC_EXECUTABLE)
+    message(FATAL_ERROR "nvcc not found")
+endif()
+
 #
 # Import torch cmake configuration.
 # Torch also imports CUDA (and partially HIP) languages with some customizations,