ARG ROCM_VERSION="5.7.3"

FROM wtongze/rocm-base:${ROCM_VERSION}

ARG HIP_ARCH="gfx803;gfx900"
ARG BLENDER_BRANCH="v3.6.15"
ARG DEBIAN_FRONTEND="noninteractive"

RUN sudo apt update && \
sudo apt install -y python3 git git-lfs

WORKDIR /home/rocm
RUN git clone --depth 1 https://projects.blender.org/blender/blender.git -b ${BLENDER_BRANCH} && \
mkdir blender/build

WORKDIR /home/rocm/blender
RUN ./build_files/build_environment/install_linux_packages.py
RUN ./build_files/utils/make_update.py --use-linux-libraries

RUN sudo apt clean && sudo rm -rf /var/lib/apt/lists/*

# > Ref: https://github.com/xuhuisheng/rocm-gfx803/blob/master/blender.md
RUN sed -i 's/__noinline__/__inline__/g' intern/cycles/kernel/device/hip/compat.h
RUN sed -i 's/major >= 9/major >= 8/g' intern/cycles/device/hip/util.h

WORKDIR /home/rocm/blender/build
RUN cmake \
-DWITH_CYCLES_HIP_BINARIES=ON \
-DCYCLES_HIP_BINARIES_ARCH=${HIP_ARCH} \
..
RUN make -j$(nproc)
RUN cpack -G TXZ

RUN cp $(ls -l blender*.tar.xz | cut -d' ' -f9 | tr -d '\n') /home/rocm/blender.tar.xz

CMD ["bash", "-l"]