diff --git a/docker/Dockerfile b/docker/Dockerfile index 149f6875b..71cef521b 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -320,7 +320,7 @@ WORKDIR /workspace # Build DeepGEMM wheel # Default moved here from tools/install_deepgemm.sh for centralized version management -ARG DEEPGEMM_GIT_REF=594953acce41793ae00a1233eb516044d604bcb6 +ARG DEEPGEMM_GIT_REF=477618cd51baffca09c4b0b87e97c03fe827ef03 COPY tools/install_deepgemm.sh /tmp/install_deepgemm.sh RUN --mount=type=cache,target=/root/.cache/uv \ mkdir -p /tmp/deepgemm/dist && \ diff --git a/docker/versions.json b/docker/versions.json index b4bc76d98..6277e0b6f 100644 --- a/docker/versions.json +++ b/docker/versions.json @@ -50,7 +50,7 @@ "default": "cuda" }, "DEEPGEMM_GIT_REF": { - "default": "594953acce41793ae00a1233eb516044d604bcb6" + "default": "477618cd51baffca09c4b0b87e97c03fe827ef03" }, "PPLX_COMMIT_HASH": { "default": "12cecfd" diff --git a/tools/install_deepgemm.sh b/tools/install_deepgemm.sh index c4028e8a7..1c316ee78 100755 --- a/tools/install_deepgemm.sh +++ b/tools/install_deepgemm.sh @@ -6,7 +6,7 @@ set -e # Default values DEEPGEMM_GIT_REPO="https://github.com/deepseek-ai/DeepGEMM.git" -DEEPGEMM_GIT_REF="0f5f2662027f0db05d4e3f6a94e56e2d8fc45c51" +DEEPGEMM_GIT_REF="477618cd51baffca09c4b0b87e97c03fe827ef03" WHEEL_DIR="" # Parse command line arguments