From f5a19920d7705175b4d788a430f8bc4e5717129d Mon Sep 17 00:00:00 2001 From: scott Date: Thu, 2 Apr 2026 21:46:56 -0400 Subject: [PATCH] Fix all ROCm 6.2 hipBLAS API incompatibilities ROCm 6.2 hipblasGemmEx uses hipblasDatatype_t for the compute type arg, not hipblasComputeType_t (that came in ROCm 7). Map compute type and HIPBLAS_COMPUTE_* constants down to the old hipblasDatatype_t/HIPBLAS_R_* equivalents. Also fix HIPBLAS_R_16B (not 16BF) and add rocrand-dev. Co-Authored-By: Claude Sonnet 4.6 --- Dockerfile | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 5fc82cb..c8f00e1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ git cmake cmake-curses-gui \ libopenblas-dev wget \ python3-dev python3-pip \ - hipblas-dev miopen-hip-dev rocthrust-dev hiprand-dev hipcub-dev \ + hipblas-dev miopen-hip-dev rocthrust-dev hiprand-dev hipcub-dev rocrand-dev \ && rm -rf /var/lib/apt/lists/* # Build oneDNN as a static library (CPU fallback / conv ops) @@ -42,11 +42,17 @@ RUN git clone --branch v4.0.0 --depth 1 \ 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_16BF/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} \