Skip to content

GitLab

Unverified Commit 9360d34f authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

update to latest deepgemm for dsv3.2 (#25871) 


Signed-off-by: default avataryoukaichao <youkaichao@gmail.com>
parent 1b67b046
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docker/Dockerfile
View file @ 9360d34f
......@@ -456,7 +456,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
ARG DEEPGEMM_GIT_REF
COPY tools/install_deepgemm.sh /tmp/install_deepgemm.sh
RUN --mount=type=cache,target=/root/.cache/uv \
VLLM_DOCKER_BUILD_CONTEXT=1 /tmp/install_deepgemm.sh --cuda-version "${CUDA_VERSION}" ${DEEPGEMM_GIT_REF:+--ref "$DEEPGEMM_GIT_REF"}
VLLM_DOCKER_BUILD_CONTEXT=1 TORCH_CUDA_ARCH_LIST="9.0a 10.0a" /tmp/install_deepgemm.sh --cuda-version "${CUDA_VERSION}" ${DEEPGEMM_GIT_REF:+--ref "$DEEPGEMM_GIT_REF"}
COPY tools/install_gdrcopy.sh install_gdrcopy.sh
RUN set -eux; \
......
tools/install_deepgemm.sh
View file @ 9360d34f
......@@ -6,7 +6,7 @@ set -e
# Default values
DEEPGEMM_GIT_REPO="https://github.com/deepseek-ai/DeepGEMM.git"
DEEPGEMM_GIT_REF="ea9c5d9270226c5dd7a577c212e9ea385f6ef048"
DEEPGEMM_GIT_REF="594953acce41793ae00a1233eb516044d604bcb6"
# Parse command line arguments
while [[ $# -gt 0 ]]; do
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment