Compare commits
3 Commits
5a7545a8c3
...
rocm7-dev
| Author | SHA1 | Date | |
|---|---|---|---|
| 473cf730d4 | |||
| c749c974e7 | |||
| 80ffcdd5da |
@@ -4,6 +4,7 @@ on:
|
|||||||
push:
|
push:
|
||||||
branches:
|
branches:
|
||||||
- main
|
- main
|
||||||
|
- rocm7-dev
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
build-and-push:
|
build-and-push:
|
||||||
@@ -28,7 +29,7 @@ jobs:
|
|||||||
context: .
|
context: .
|
||||||
push: true
|
push: true
|
||||||
tags: |
|
tags: |
|
||||||
git.sdgarren.com/scott/rocm-faster-whisper:latest
|
git.sdgarren.com/scott/rocm-faster-whisper:${{ gitea.ref_name }}
|
||||||
git.sdgarren.com/scott/rocm-faster-whisper:${{ gitea.sha }}
|
git.sdgarren.com/scott/rocm-faster-whisper:${{ gitea.sha }}
|
||||||
cache-from: type=registry,ref=git.sdgarren.com/scott/rocm-faster-whisper:buildcache
|
cache-from: type=registry,ref=git.sdgarren.com/scott/rocm-faster-whisper:buildcache
|
||||||
cache-to: type=registry,ref=git.sdgarren.com/scott/rocm-faster-whisper:buildcache,mode=max
|
cache-to: type=registry,ref=git.sdgarren.com/scott/rocm-faster-whisper:buildcache,mode=max
|
||||||
|
|||||||
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