x

2026-06-05 09:55:17 +08:00
parent 09175d61e7
commit 805acbf0a1
1 changed files with 2 additions and 1 deletions
@@ -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/*

 # ===========================================================================