From 3797117a249528f4b204d3025207e0049827e0fb Mon Sep 17 00:00:00 2001
From: Zach Dunton <fenfir@gmail.com>
Date: Mon, 15 Jan 2024 22:35:33 +0100
Subject: [PATCH] Dockerfile changes to build for ROCm

---
 Dockerfile | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index b5217da0ead2..764710d01192 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,8 @@
-ARG GO_VERSION=1.21-bullseye
+ARG GO_VERSION=1.21-bookworm
 ARG IMAGE_TYPE=extras
 # extras or core
 
+FROM rocm/dev-ubuntu-22.04:6.0-complete as rocm
 
 FROM golang:$GO_VERSION as requirements-core
 
@@ -40,6 +41,11 @@ RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \
     ; fi
 ENV PATH /usr/local/cuda/bin:${PATH}
 
+# HipBLAS requirements
+COPY --from=rocm /opt/rocm /opt/rocm
+ENV PATH /opt/rocm/bin:${PATH}
+ENV ROCM_PATH=/opt/rocm
+
 # OpenBLAS requirements and stable diffusion
 RUN apt-get install -y \
     libopenblas-dev \
@@ -66,7 +72,7 @@ RUN curl https://repo.anaconda.com/pkgs/misc/gpgkeys/anaconda.asc | gpg --dearmo
     apt-get install -y conda && apt-get clean
 
 ENV PATH="/root/.cargo/bin:${PATH}"
-RUN pip install --upgrade pip
+RUN apt-get install -y python3-pip && apt-get clean
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
 RUN apt-get install -y espeak-ng espeak && apt-get clean