Skip to content

Commit 659880b

Browse files
authored
Merge pull request #602 from ROCm/deepseek_085_fixbug
add expert mask for fused_moe when running in epmoe mode
2 parents 22e44cb + 4bcb38e commit 659880b

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

vllm/model_executor/layers/fused_moe/rocm_aiter_fused_moe.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -249,6 +249,7 @@ def rocm_aiter_asm_moe_impl(
249249
w2_scale=fc2_scale,
250250
quant_type=quant_type,
251251
activation=activation,
252+
expert_mask=expert_mask,
252253
)
253254

254255

0 commit comments

Comments
 (0)