docker: fix offload-arch in rocm

devitocodes · Jan 25, 2024 · 735eaa5 · 735eaa5
1 parent a04059e
commit 735eaa5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/docker/Dockerfile.amd b/docker/Dockerfile.amd
@@ -45,6 +45,9 @@ ENV ROCM_HOME=/opt/rocm \
     UCX_HOME=/opt/ucx \
     OMPI_HOME=/opt/ompi
 
+# Until rocm base has it fixed
+RUN ln -s /opt/rocm/llvm/bin/offload-arch /opt/rocm/bin/offload-arch
+
 # Install tmpi
 RUN curl https://raw.githubusercontent.com/Azrael3000/tmpi/master/tmpi -o /usr/local/bin/tmpi