From 805acbf0a1df56f10298351b1a3d815b512ddb0b Mon Sep 17 00:00:00 2001
From: chenjw28 <792430652@qq.com>
Date: Fri, 5 Jun 2026 09:55:17 +0800
Subject: [PATCH] x

---
 docker/Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 513480a..1dcfa12 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -128,7 +128,8 @@ RUN python${PYTHON_VER} -m venv ${VENV} && \
 RUN apt-get update && apt-get install -y --no-install-recommends \
     hipblas-dev hiprand-dev hipsparse-dev hipsparselt-dev \
     hipsolver-dev hipcub-dev rocprim-dev rocthrust-dev \
-    rocblas-dev rocrand-dev hipfft-dev hipblaslt && \
+    rocblas-dev rocrand-dev hipfft-dev hipblaslt \
+    rocm-cmake && \
     apt-get clean && rm -rf /var/lib/apt/lists/*
 
 # ===========================================================================