EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[ENV, ROCm] update environment settings (#1939)

Browse Source
This commit is contained in:
HAI
2024-11-07 18:24:36 -08:00
committed by GitHub
parent 67c424cce3
commit d32fba2a4d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile.rocm
View File

@@ -8,7 +8,7 @@ FROM $BASE_IMAGE AS base
USER root USER root
WORKDIR /sgl-workspace WORKDIR /sgl-workspace
ARG BUILD_TYPE=all
ARG SGL_REPO="https://github.com/sgl-project/sglang" ARG SGL_REPO="https://github.com/sgl-project/sglang"
ENV SGL_DEFAULT="main" ENV SGL_DEFAULT="main"
ARG SGL_BRANCH=${SGL_DEFAULT} ARG SGL_BRANCH=${SGL_DEFAULT}
@@ -41,5 +41,7 @@ ENV VLLM_FP8_PADDING=1
ENV VLLM_FP8_ACT_PADDING=1 ENV VLLM_FP8_ACT_PADDING=1
ENV VLLM_FP8_WEIGHT_PADDING=1 ENV VLLM_FP8_WEIGHT_PADDING=1
ENV VLLM_FP8_REDUCE_CONV=1 ENV VLLM_FP8_REDUCE_CONV=1
ENV TORCHINDUCTOR_MAX_AUTOTUNE=1
ENV TORCHINDUCTOR_MAX_AUTOTUNE_POINTWISE=1
CMD ["/bin/bash"] CMD ["/bin/bash"]