Using ROCm6.3 release image as a base (#308)

ROCm · Dec 6, 2024 · 2b17421 · 2b17421
1 parent da8f61a
commit 2b17421
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Dockerfile.base b/Dockerfile.base
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:6.2.4-complete
+ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:6.3-complete
 FROM ${BASE_IMAGE} AS base
 
 ENV PATH=/opt/rocm/llvm/bin:$PATH
@@ -30,8 +30,8 @@ RUN pip install -U packaging cmake ninja wheel setuptools Cython
 
 FROM base AS build_hipblaslt
 ARG HIPBLASLT_BRANCH="507a649"
-# Unset the following on ROCm6.3+
-ARG LEGACY_HIPBLASLT_OPTION="--legacy_hipblas_direct"
+# Set to "--legacy_hipblas_direct" for ROCm<=6.2
+ARG LEGACY_HIPBLASLT_OPTION=
 RUN git clone https://github.com/ROCm/hipBLASLt
 RUN cd hipBLASLt \
     && git checkout ${HIPBLASLT_BRANCH} \

diff --git a/Dockerfile.rocm b/Dockerfile.rocm
@@ -3,7 +3,7 @@ ARG REMOTE_VLLM="0"
 ARG USE_CYTHON="0"
 ARG BUILD_RPD="1"
 ARG COMMON_WORKDIR=/app
-ARG BASE_IMAGE=rocm/vllm-dev:base_ubuntu22.04_py3.12_ROCm6.2.4_hipblaslt0.11_torch2.6
+ARG BASE_IMAGE=rocm/vllm-dev:base_ubuntu22.04_py3.12_ROCm6.3_hipblaslt0.11_torch2.6
 
 FROM ${BASE_IMAGE} AS base