FROM ubuntu:22.04
ARG ROCM_VERSION="5.7.1"
ARG ROCBLAS_BRANCH="release/rocm-rel-5.7"
ARG DEBIAN_FRONTEND="noninteractive"
RUN apt-get update \
&& apt-get install -y wget git \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /root
RUN wget -O amdgpu-install.deb https://repo.radeon.com/amdgpu-install/${ROCM_VERSION}/ubuntu/jammy/amdgpu-install_5.7.50701-1_all.deb
RUN dpkg -i amdgpu-install.deb
RUN amdgpu-install -y --no-dkms --usecase=rocm
RUN git clone -b ${ROCBLAS_BRANCH} https://github.com/ROCmSoftwarePlatform/rocBLAS.git
WORKDIR /root/rocBLAS
RUN rm -rf library/src/blas3/Tensile/Logic/asm_full/r9nano/

# gfx803 as single target will cause error
RUN sed -i -e 's/rmake.py --install/rmake.py -a="gfx803;gfx900" --install/g' install.sh

RUN ./install.sh -id --cmake_install
WORKDIR /root

## need to fix broken version
CMD [ "/bin/bash" ]