Bladeren bron

docker: apply AMD patch in the dockerfile (#777)

AlpinDale 4 maanden geleden
bovenliggende
commit
4d781b22d3
1 gewijzigde bestanden met toevoegingen van 5 en 0 verwijderingen
  1. 5 0
      Dockerfile.rocm

+ 5 - 0
Dockerfile.rocm

@@ -131,6 +131,11 @@ COPY . .
 RUN --mount=type=cache,target=/root/.cache/pip \
     python3 -m pip install --upgrade numba scipy huggingface-hub[cli]
 
+
+# Clang needs to be patched for ROCm
+RUN chmod +x patches/amd.patch
+RUN patch -p1 < patches/amd.patch
+
 # Workaround for ray >= 2.10.0
 ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=1
 # Silences the HF Tokenizers warning