diff --git a/Dockerfile b/Dockerfile index 9389591..524e10d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -ARG ROCM_VERSION=6.2 +ARG ROCM_VERSION=7.0 ARG GPU_ARCH="gfx1030;gfx1031" ARG ONEDNN_VERSION=3.1.1 @@ -40,20 +40,6 @@ RUN git clone --branch v4.0.0 --depth 1 \ cd /build/CTranslate2 && \ git submodule update --init --recursive && \ 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 && \ cmake -DCMAKE_INSTALL_PREFIX=${CTRANSLATE2_ROOT} \ -DCMAKE_PREFIX_PATH=/opt/rocm \