x
This commit is contained in:
+2
-2
@@ -113,7 +113,7 @@ ENV VLLM_TARGET_DEVICE="rocm"
|
|||||||
ENV PYTORCH_ROCM_ARCH="gfx1201"
|
ENV PYTORCH_ROCM_ARCH="gfx1201"
|
||||||
ENV HIP_ARCHITECTURES="gfx1201"
|
ENV HIP_ARCHITECTURES="gfx1201"
|
||||||
ENV AMDGPU_TARGETS="gfx1201"
|
ENV AMDGPU_TARGETS="gfx1201"
|
||||||
ENV MAX_JOBS="4"
|
ENV MAX_JOBS="32"
|
||||||
|
|
||||||
# --- FIX FOR SEGFAULT ---
|
# --- FIX FOR SEGFAULT ---
|
||||||
# We force the Host Compiler (CC/CXX) to be the ROCm Clang, not Fedora GCC.
|
# We force the Host Compiler (CC/CXX) to be the ROCm Clang, not Fedora GCC.
|
||||||
@@ -145,7 +145,7 @@ RUN cmake -S . \
|
|||||||
-DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
|
-DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
|
||||||
-DCMAKE_CXX_COMPILER=/opt/rocm/llvm/bin/clang++ \
|
-DCMAKE_CXX_COMPILER=/opt/rocm/llvm/bin/clang++ \
|
||||||
&& \
|
&& \
|
||||||
make -j$(nproc) && \
|
make -j32 && \
|
||||||
python -m pip install --no-cache-dir . --no-build-isolation --no-deps
|
python -m pip install --no-cache-dir . --no-build-isolation --no-deps
|
||||||
|
|
||||||
# 8. Final Cleanup & Runtime
|
# 8. Final Cleanup & Runtime
|
||||||
|
|||||||
Reference in New Issue
Block a user