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