From 37482768f399566c01ebe39a023d35b9cdbc08f5 Mon Sep 17 00:00:00 2001
From: "Yaxun (Sam) Liu" <yaxun.liu@amd.com>
Date: Thu, 24 Apr 2025 11:42:41 -0400
Subject: [PATCH] fix rocm-6.4.0 build

add rocm-cmake since device library now depends on it.
---
 rocm/build.sh | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/rocm/build.sh b/rocm/build.sh
index e804c2e..48abd4f 100755
--- a/rocm/build.sh
+++ b/rocm/build.sh
@@ -36,6 +36,20 @@ ${OPT}/infra/bin/ce_install install "clang-rocm ${VERSION}"
 COMP=${OPT}/clang-rocm-${VERSION}
 DEST=${OPT}/libs/rocm/${VERSION}
 
+# rocm-cmake
+if (( ROCM_MAJOR_MINOR >= 604 )); then
+  rm -rf rocm-cmake-${ROCM_VERSION}
+  curl -sL https://github.com/ROCm/rocm-cmake/archive/refs/tags/${ROCM_VERSION}.tar.gz | tar xz
+  pushd rocm-cmake-${ROCM_VERSION}
+  cmake -S. -Bbuild \
+    -GNinja \
+    -DCMAKE_PREFIX_PATH="${COMP};${DEST}" \
+    -DCMAKE_INSTALL_PREFIX="${DEST}"
+  ninja -C build
+  ninja -C build install
+  popd
+fi
+
 rm -rf llvm-project-$ROCM_VERSION ROCm-Device-Libs-$ROCM_VERSION ROCm-CompilerSupport-$ROCM_VERSION HIPCC-$ROCM_VERSION
 if (( ROCM_MAJOR_MINOR < 601 )); then
     curl -sL https://github.com/ROCm/ROCm-Device-Libs/archive/refs/tags/${ROCM_VERSION}.tar.gz | tar xz