From 98e71a495457251f26d46af6b0614831bf033269 Mon Sep 17 00:00:00 2001 From: Zhuohan Li Date: Wed, 15 Oct 2025 17:01:03 -0700 Subject: [PATCH] enable all Signed-off-by: Zhuohan Li --- .../layers/moe/grouped_gemm_no_abstraction.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vllm/model_executor/layers/moe/grouped_gemm_no_abstraction.py b/vllm/model_executor/layers/moe/grouped_gemm_no_abstraction.py index 128201f6f34da..d73daaa846a61 100644 --- a/vllm/model_executor/layers/moe/grouped_gemm_no_abstraction.py +++ b/vllm/model_executor/layers/moe/grouped_gemm_no_abstraction.py @@ -351,6 +351,6 @@ def run_triton_group_gemm_masked_bf16( # run_batched_deepgemm_masked_fp8(512, 8, 1024, 512) run_batched_deepgemm_contiguous_bf16(512, 8, 1024, 512) -# run_batched_deepgemm_masked_bf16(512, 8, 1024, 512) -# run_triton_group_gemm_contiguous_bf16(512, 8, 1024, 512, 4) -# run_triton_group_gemm_masked_bf16(512, 8, 1024, 512) +run_batched_deepgemm_masked_bf16(512, 8, 1024, 512) +run_triton_group_gemm_contiguous_bf16(512, 8, 1024, 512, 4) +run_triton_group_gemm_masked_bf16(512, 8, 1024, 512)