From 80e254834de9c3c34eaca02d8880e952b3daf344 Mon Sep 17 00:00:00 2001 From: James Whedbee Date: Wed, 20 Mar 2024 16:05:03 -0500 Subject: [PATCH] [Bugfix] Fix ROCm support in CMakeLists.txt (#3534) --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 150fcebe..66842e68 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -51,7 +51,7 @@ 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) +if (CUDA_FOUND AND NOT NVCC_EXECUTABLE) message(FATAL_ERROR "nvcc not found") endif()