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

update ROCm docker for layernorm kernel optimization (#2885)

Browse Source 
Co-authored-by: wunhuang <wunhuang@amd.com>
This commit is contained in:
kk
2025-01-14 16:59:43 +08:00
committed by GitHub
parent c19d84829c
commit b8cd09f27a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile.rocm
View File

@@ -2,7 +2,7 @@
# docker build --build-arg SGL_BRANCH=v0.4.1.post5 -t v0.4.1.post5-rocm620 -f Dockerfile.rocm .
# default base image
ARG BASE_IMAGE="rocmshared/vllm-rocm:20250113-tuned-elementwise"
ARG BASE_IMAGE="rocmshared/vllm-rocm:20250114-tuned-elementwise-layernorm"
FROM $BASE_IMAGE AS base
USER root