From 6e8c3751c504b4f4ef9a1dda9bd10fc0b404bd3b Mon Sep 17 00:00:00 2001
From: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
Date: Tue, 26 Mar 2024 15:56:15 +0000
Subject: [PATCH] Building hipblaslt

---
 Dockerfile.rocm | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 32ab7cf96fc59..a57d922df5762 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -27,6 +27,9 @@ ARG BUILD_CUPY="1"
 # whether to build triton on rocm
 ARG BUILD_TRITON="1"
 
+# whether to build hipBLASLt
+ARG BUILD_HIPBLASLT="1"
+
 # Install some basic utilities
 RUN apt-get update && apt-get install -y \
     sqlite3 libsqlite3-dev libfmt-dev \
@@ -83,6 +86,16 @@ RUN if [ "$BUILD_CUPY" = "1" ]; then \
     && cd ..; \
     fi
 
+RUN if [ "$BUILD_HIPBLASLT" = "1" ] ; then \
+    git clone -b develop https://github.com/ROCm/hipBLASLt \
+    && export GTest_DIR="/usr/local/lib/cmake/GTest/" \
+    && cd hipBLASLt \
+    && ./install.sh -idc --architecture 'gfx90a;gfx942' \
+    && cd ../ && rm -rf hipBLASLt \
+    && sed -i 's/, hipblaslt-dev \(.*\), hipcub-dev/, hipcub-dev/g' /var/lib/dpkg/status \
+    && sed -i 's/, hipblaslt \(.*\), hipfft/, hipfft/g' /var/lib/dpkg/status; \
+    fi
+
 # build triton
 RUN if [ "$BUILD_TRITON" = "1" ]; then \
     mkdir -p libs \