diff --git a/codegen/CMakeLists.txt b/codegen/CMakeLists.txt
index 8df808b08c..75da93f0b8 100644
--- a/codegen/CMakeLists.txt
+++ b/codegen/CMakeLists.txt
@@ -16,6 +16,10 @@ set(CMAKE_INSTALL_PREFIX "/opt/rocm" CACHE PATH "")
 list(APPEND CMAKE_PREFIX_PATH /opt/rocm /opt/rocm/llvm $ENV{ROCM_PATH} $ENV{HIP_PATH})
 list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
 
+if(NOT ROCmCMakeBuildTools_DIR)
+  set(ROCmCMakeBuildTools_DIR "/opt/rocm/lib/cmake/ROCmCMakeBuildTools")
+endif()
+
 find_package(ROCmCMakeBuildTools REQUIRED)
     
 include(ROCMInstallTargets)