Improve moe reduce sum kernel performance (#2705)

Co-authored-by: wunhuang <wunhuang@amd.com>
2025-01-02 17:11:06 +08:00
parent a4d6d6f1dd
commit 148254d4db
2 changed files with 12 additions and 6 deletions
--- a/docker/Dockerfile.rocm
+++ b/docker/Dockerfile.rocm
@@ -2,7 +2,7 @@
 #   docker build --build-arg SGL_BRANCH=v0.4.1.post3 -t v0.4.1.post3-rocm620 -f Dockerfile.rocm .

 # default base image
-ARG BASE_IMAGE="rocm/vllm-dev:20241031-tuned"
+ARG BASE_IMAGE="rocmshared/vllm-rocm:20241031-tuned"

 FROM $BASE_IMAGE AS base
 USER root