updateing HIP path for 6.0 release

ROCm · Feb 27, 2024 · 0cdfd46 · 0cdfd46
1 parent 596f7e1
commit 0cdfd46
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -74,7 +74,7 @@ if(USE_ROCM)
   # Find the HIP package, set the HIP paths, load the HIP CMake.
   include(cmake/LoadHIP.cmake)
   if(NOT PYTORCH_FOUND_HIP)
-    set(USE_ROCM OFF)
+    message (FATAL_ERROR "HIP NOT Found")
   endif()
 
   if(CMAKE_VERSION VERSION_LESS 3.21.0)

diff --git a/cmake/LoadHIP.cmake b/cmake/LoadHIP.cmake
@@ -129,7 +129,7 @@ else()
 endif()
 
 # Add HIP to the CMAKE Module Path
-set(CMAKE_MODULE_PATH ${HIP_PATH}/cmake ${CMAKE_MODULE_PATH})
+set(CMAKE_MODULE_PATH ${ROCM_PATH}/lib/cmake/hip ${CMAKE_MODULE_PATH})
 
 # Disable Asserts In Code (Can't use asserts on HIP stack.)
 add_definitions(-DNDEBUG)