From e70b3860d36b279f0ebadfd63cdd96649ed93458 Mon Sep 17 00:00:00 2001
From: mudler <mudler@localai.io>
Date: Wed, 14 Feb 2024 23:38:12 +0100
Subject: [PATCH] ci: build hipblas images

---
 .github/workflows/image-pr.yml |  8 ++++++++
 .github/workflows/image.yml    | 32 ++++++++++++++++++++++++++++++++
 Dockerfile                     |  2 ++
 3 files changed, 42 insertions(+)

diff --git a/.github/workflows/image-pr.yml b/.github/workflows/image-pr.yml
index ae8bd070a12..793353dbe95 100644
--- a/.github/workflows/image-pr.yml
+++ b/.github/workflows/image-pr.yml
@@ -51,6 +51,14 @@ jobs:
             image-type: 'extras'
             runs-on: 'arc-runner-set'
             base-image: "ubuntu:22.04"
+          - build-type: 'hipblas'
+            platforms: 'linux/amd64'
+            tag-latest: 'false'
+            tag-suffix: '-hipblas'
+            ffmpeg: 'false'
+            image-type: 'extras'
+            base-image: "rocm/dev-ubuntu-22.04:6.0-complete"
+            runs-on: 'ubuntu-latest'
   core-image-build:
     uses: ./.github/workflows/image_build.yml
     with:
diff --git a/.github/workflows/image.yml b/.github/workflows/image.yml
index ac61deeca6e..ee509a9d771 100644
--- a/.github/workflows/image.yml
+++ b/.github/workflows/image.yml
@@ -103,6 +103,22 @@ jobs:
             image-type: 'extras'
             base-image: "ubuntu:22.04"
             runs-on: 'arc-runner-set'
+          - build-type: 'hipblas'
+            platforms: 'linux/amd64'
+            tag-latest: 'false'
+            tag-suffix: '-hipblas-ffmpeg'
+            ffmpeg: 'true'
+            image-type: 'extras'
+            base-image: "rocm/dev-ubuntu-22.04:6.0-complete"
+            runs-on: 'ubuntu-latest'
+          - build-type: 'hipblas'
+            platforms: 'linux/amd64'
+            tag-latest: 'false'
+            tag-suffix: '-hipblas'
+            ffmpeg: 'false'
+            image-type: 'extras'
+            base-image: "rocm/dev-ubuntu-22.04:6.0-complete"
+            runs-on: 'ubuntu-latest'
   core-image-build:
     uses: ./.github/workflows/image_build.yml
     with:
@@ -124,6 +140,22 @@ jobs:
     strategy:
       matrix:
         include:
+          - build-type: 'hipblas'
+            platforms: 'linux/amd64'
+            tag-latest: 'false'
+            tag-suffix: '-hipblas-ffmpeg-core'
+            ffmpeg: 'true'
+            image-type: 'core'
+            base-image: "rocm/dev-ubuntu-22.04:6.0-complete"
+            runs-on: 'ubuntu-latest'
+          - build-type: 'hipblas'
+            platforms: 'linux/amd64'
+            tag-latest: 'false'
+            tag-suffix: '-hipblas-core'
+            ffmpeg: 'false'
+            image-type: 'core'
+            base-image: "rocm/dev-ubuntu-22.04:6.0-complete"
+            runs-on: 'ubuntu-latest'
           - build-type: ''
             platforms: 'linux/amd64'
             tag-latest: 'false'
diff --git a/Dockerfile b/Dockerfile
index 15e0cec0120..92b6a3f45da 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -78,6 +78,8 @@ RUN curl https://repo.anaconda.com/pkgs/misc/gpgkeys/anaconda.asc | gpg --dearmo
 
 ENV PATH="/root/.cargo/bin:${PATH}"
 RUN apt-get install -y python3-pip && apt-get clean
+RUN pip install --upgrade pip
+
 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