From 51f4a17fd33b36fba7c884a5bdaba66ef6cd50d6 Mon Sep 17 00:00:00 2001
From: "Yuhsiang M. Tsai" <yhmtsai@gmail.com>
Date: Sat, 21 Nov 2020 23:01:25 +0100
Subject: [PATCH] use HIP_PATH to add the module path

---
 CMakeLists.txt | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index a2b09e79..282cc85b 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -31,10 +31,17 @@ else()
   set(CMAKE_INSTALL_PREFIX "/opt/rocm" CACHE PATH "Install path prefix, prepended onto install directories")
 endif()
 
+# Get HIP_PATH
+if( NOT DEFINED ENV{HIP_PATH} )
+  set( HIP_PATH "/opt/rocm/hip" )
+else()
+  set( HIP_PATH $ENV{HIP_PATH} )
+endif()
+
 # CMake modules
 list(APPEND CMAKE_MODULE_PATH
      ${CMAKE_CURRENT_SOURCE_DIR}/cmake
-     /opt/rocm/hip/cmake
+     ${HIP_PATH}/cmake
 )
 
 # Set a default build type if none was specified