Skip to content

GitLab

Unverified Commit 84eaa684 authored by Allen.Dou's avatar Allen.Dou Committed by GitHub
Browse files

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

parent 5ee14494
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
CMakeLists.txt
View file @ 84eaa684
...@@ -49,6 +49,12 @@ endif() ...@@ -49,6 +49,12 @@ endif()
# #
append_cmake_prefix_path("torch" "torch.utils.cmake_prefix_path") 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. # Import torch cmake configuration.
# Torch also imports CUDA (and partially HIP) languages with some customizations, # Torch also imports CUDA (and partially HIP) languages with some customizations,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment