Compare commits
4 Commits
5a7545a8c3
...
464c3d372c
| Author | SHA1 | Date | |
|---|---|---|---|
| 464c3d372c | |||
| 473cf730d4 | |||
| c749c974e7 | |||
| 80ffcdd5da |
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 \
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ services:
|
|||||||
build:
|
build:
|
||||||
context: .
|
context: .
|
||||||
args:
|
args:
|
||||||
ROCM_VERSION: "6.2"
|
ROCM_VERSION: "7.0"
|
||||||
GPU_ARCH: "gfx1030;gfx1031"
|
GPU_ARCH: "gfx1030;gfx1031"
|
||||||
image: rocm-faster-whisper:latest
|
image: rocm-faster-whisper:latest
|
||||||
devices:
|
devices:
|
||||||
|
|||||||
Reference in New Issue
Block a user