Upgrade to ROCm 7.0 and remove 6.2 API backport shims
The HIP patch already targets the ROCm 7 hipBLAS API (HIPBLAS_COMPUTE_*, hipDataType, HIP_R_*). The sed block was explicitly downgrading those to the deprecated ROCm 6.2 equivalents; removing it lets the patch build natively against ROCm 7. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
16
Dockerfile
16
Dockerfile
@@ -1,4 +1,4 @@
|
|||||||
ARG ROCM_VERSION=6.2
|
ARG ROCM_VERSION=7.0
|
||||||
ARG GPU_ARCH="gfx1030;gfx1031"
|
ARG GPU_ARCH="gfx1030;gfx1031"
|
||||||
ARG ONEDNN_VERSION=3.1.1
|
ARG ONEDNN_VERSION=3.1.1
|
||||||
|
|
||||||
@@ -40,20 +40,6 @@ RUN git clone --branch v4.0.0 --depth 1 \
|
|||||||
cd /build/CTranslate2 && \
|
cd /build/CTranslate2 && \
|
||||||
git submodule update --init --recursive && \
|
git submodule update --init --recursive && \
|
||||||
git apply /tmp/ct2_4.0.0_rocm.patch && \
|
git apply /tmp/ct2_4.0.0_rocm.patch && \
|
||||||
sed -i \
|
|
||||||
-e 's/hipDataType/hipblasDatatype_t/g' \
|
|
||||||
-e 's/hipblasComputeType_t/hipblasDatatype_t/g' \
|
|
||||||
-e 's/HIP_R_16F/HIPBLAS_R_16F/g' \
|
|
||||||
-e 's/HIP_R_32F/HIPBLAS_R_32F/g' \
|
|
||||||
-e 's/HIP_R_16BF/HIPBLAS_R_16B/g' \
|
|
||||||
-e 's/HIP_R_32I/HIPBLAS_R_32I/g' \
|
|
||||||
-e 's/HIP_R_8I/HIPBLAS_R_8I/g' \
|
|
||||||
-e 's/HIPBLAS_COMPUTE_16F/HIPBLAS_R_16F/g' \
|
|
||||||
-e 's/HIPBLAS_COMPUTE_32F_FAST_16BF/HIPBLAS_R_32F/g' \
|
|
||||||
-e 's/HIPBLAS_COMPUTE_32F_FAST_16F/HIPBLAS_R_32F/g' \
|
|
||||||
-e 's/HIPBLAS_COMPUTE_32F/HIPBLAS_R_32F/g' \
|
|
||||||
-e 's/HIPBLAS_COMPUTE_32I/HIPBLAS_R_32I/g' \
|
|
||||||
src/cuda2hip_macros.hpp && \
|
|
||||||
mkdir build && cd build && \
|
mkdir build && cd build && \
|
||||||
cmake -DCMAKE_INSTALL_PREFIX=${CTRANSLATE2_ROOT} \
|
cmake -DCMAKE_INSTALL_PREFIX=${CTRANSLATE2_ROOT} \
|
||||||
-DCMAKE_PREFIX_PATH=/opt/rocm \
|
-DCMAKE_PREFIX_PATH=/opt/rocm \
|
||||||
|
|||||||
Reference in New Issue
Block a user