Signed-off-by: Zifei Tong <zifeitong@gmail.com> Signed-off-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com> Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>
This commit is contained in:
@@ -320,7 +320,7 @@ WORKDIR /workspace
|
|||||||
|
|
||||||
# Build DeepGEMM wheel
|
# Build DeepGEMM wheel
|
||||||
# Default moved here from tools/install_deepgemm.sh for centralized version management
|
# 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
|
COPY tools/install_deepgemm.sh /tmp/install_deepgemm.sh
|
||||||
RUN --mount=type=cache,target=/root/.cache/uv \
|
RUN --mount=type=cache,target=/root/.cache/uv \
|
||||||
mkdir -p /tmp/deepgemm/dist && \
|
mkdir -p /tmp/deepgemm/dist && \
|
||||||
|
|||||||
@@ -50,7 +50,7 @@
|
|||||||
"default": "cuda"
|
"default": "cuda"
|
||||||
},
|
},
|
||||||
"DEEPGEMM_GIT_REF": {
|
"DEEPGEMM_GIT_REF": {
|
||||||
"default": "594953acce41793ae00a1233eb516044d604bcb6"
|
"default": "477618cd51baffca09c4b0b87e97c03fe827ef03"
|
||||||
},
|
},
|
||||||
"PPLX_COMMIT_HASH": {
|
"PPLX_COMMIT_HASH": {
|
||||||
"default": "12cecfd"
|
"default": "12cecfd"
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ set -e
|
|||||||
|
|
||||||
# Default values
|
# Default values
|
||||||
DEEPGEMM_GIT_REPO="https://github.com/deepseek-ai/DeepGEMM.git"
|
DEEPGEMM_GIT_REPO="https://github.com/deepseek-ai/DeepGEMM.git"
|
||||||
DEEPGEMM_GIT_REF="0f5f2662027f0db05d4e3f6a94e56e2d8fc45c51"
|
DEEPGEMM_GIT_REF="477618cd51baffca09c4b0b87e97c03fe827ef03"
|
||||||
WHEEL_DIR=""
|
WHEEL_DIR=""
|
||||||
|
|
||||||
# Parse command line arguments
|
# Parse command line arguments
|
||||||
|
|||||||
Reference in New Issue
Block a user