Skip to content

Commit 30a33b9

Browse files
varun-sundar-rabindranathVarun Sundar Rabindranath
andauthored
[Misc] Rev DeepEP (vllm-project#27122)
Signed-off-by: Varun Sundar Rabindranath <vsundarr@redhat.com> Co-authored-by: Varun Sundar Rabindranath <vsundarr@redhat.com>
1 parent 7c57254 commit 30a33b9

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tools/ep_kernels/install_python_libraries.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ popd
119119

120120
# build and install deepep, require pytorch installed
121121
pushd $WORKSPACE
122-
clone_repo "https://github.com/deepseek-ai/DeepEP" "DeepEP" "setup.py" "e3908bf"
122+
clone_repo "https://github.com/deepseek-ai/DeepEP" "DeepEP" "setup.py" "73b6ea4"
123123
cd DeepEP
124124
export NVSHMEM_DIR=$WORKSPACE/nvshmem_install
125125
$PIP_CMD install --no-build-isolation -vvv -e .

vllm/model_executor/layers/fused_moe/deepep_ll_prepare_finalize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ class DeepEPLLPrepareAndFinalize(mk.FusedMoEPrepareAndFinalize):
5252
# specific hidden sizes.
5353
# NOTE: Keep this list sorted, maybe_roundup_layer_hidden_size depends
5454
# on it.
55-
SUPPORTED_HIDDEN_SIZES = [2048, 2560, 4096, 5120, 6144, 7168, 8192]
55+
SUPPORTED_HIDDEN_SIZES = [2048, 2560, 3072, 4096, 5120, 6144, 7168, 8192]
5656

5757
@staticmethod
5858
def maybe_roundup_layer_hidden_size(hidden_size: int) -> int:

0 commit comments

Comments
 (0)