Copy config files for MI300X to support in virtualized environments (#3505)

This commit is contained in:
Jesse Lopez
2025-02-14 09:23:32 -08:00
committed by GitHub
parent 4e23c961e8
commit 6ce6eabbcc

View File

@@ -58,6 +58,11 @@ RUN git clone ${ATER_REPO} \
&& git submodule update --init --recursive \
&& PREBUILD_KERNELS=1 GPU_ARCHS=gfx942 python3 setup.py develop
# Copy config files to support MI300X in virtualized environments (MI300X_VF). Symlinks will not be created in image build.
RUN find /sgl-workspace/sglang/python/sglang/srt/layers/quantization/configs/ \
/sgl-workspace/sglang/python/sglang/srt/layers/moe/fused_moe_triton/configs/ \
-type f -name '*MI300X*' | xargs -I {} sh -c 'vf_config=$(echo "$1" | sed "s/MI300X/MI300X_VF/"); cp "$1" "$vf_config"' -- {}
# Performance environment variable.
ENV HIP_FORCE_DEV_KERNARG=1