From b280b7fa094af19be81a6e1727d18c0b83dbcb6f Mon Sep 17 00:00:00 2001
From: Mathieu Broillet <mathieu@broillet.ch>
Date: Wed, 28 Aug 2024 16:02:15 +0200
Subject: [PATCH] switch to -complete docker image

---
 .github/workflows/build-and_publish-prebuilts-whl.yml | 6 ++++--
 prebuilts/Dockerfile                                  | 5 ++++-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/build-and_publish-prebuilts-whl.yml b/.github/workflows/build-and_publish-prebuilts-whl.yml
index 93e46bb..47b86c7 100644
--- a/.github/workflows/build-and_publish-prebuilts-whl.yml
+++ b/.github/workflows/build-and_publish-prebuilts-whl.yml
@@ -17,8 +17,10 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - name: Delete huge unnecessary tools folder
-        run: rm -rf /opt/hostedtoolcache
+      - name: Free Disk Space (Ubuntu)
+        uses: jlumbroso/free-disk-space@main
+        with:
+          tool-cache: false
 
       # Step 1: Checkout the repository
       - name: Checkout repository
diff --git a/prebuilts/Dockerfile b/prebuilts/Dockerfile
index b1da906..f088e03 100644
--- a/prebuilts/Dockerfile
+++ b/prebuilts/Dockerfile
@@ -1,5 +1,6 @@
 ARG ROCM_VERSION=6.1.2
-FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION}
+FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION}-complete
+# also works without -complete for specific arch (tested on gfx1030)
 
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 \
@@ -10,6 +11,8 @@ ENV BNB_GPU_TARGETS="gfx803;gfx900;gfx906;gfx908;gfx90a;gfx1010;gfx1030;gfx1100;
 # For LLAMA
 ENV GPU_TARGETS="gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101"
 
+ENV CC=/opt/rocm/llvm/bin/clang
+ENV CXX=/opt/rocm/llvm/bin/clang++
 
 WORKDIR /tmp