From 7a40e4f4a66ddf595ee45a886ddb1c75f68dee4a Mon Sep 17 00:00:00 2001
From: Rain Jiang <96632942+rainj-me@users.noreply.github.com>
Date: Mon, 8 Sep 2025 16:24:55 -0700
Subject: [PATCH] fix the cutlass moe tests (#10182)

---
 python/sglang/test/test_cutlass_moe.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/python/sglang/test/test_cutlass_moe.py b/python/sglang/test/test_cutlass_moe.py
index f6bc2b0b2..56f276c81 100755
--- a/python/sglang/test/test_cutlass_moe.py
+++ b/python/sglang/test/test_cutlass_moe.py
@@ -22,7 +22,7 @@ def calc_diff(x, y):
 
 def get_model_config(tp_size: int):
     config = AutoConfig.from_pretrained(
-        "deepseek-ai/deepseek-R1", trust_remote_code=True
+        "deepseek-ai/Deepseek-R1", trust_remote_code=True
     )
     E = config.n_routed_experts
     topk = config.num_experts_per_tok
@@ -163,11 +163,10 @@ def run_test(tp_size, batch_size, model_config, check=False):
 
     moe_runner_config = MoeRunnerConfig(
         num_experts=E,
-        topk=topk,
+        top_k=topk,
         hidden_size=H,
-        shard_intermediate_size=I,
-        dtype=dtype,
-        block_shape=block_shape,
+        intermediate_size_per_partition=I,
+        params_dtype=dtype,
         activation="silu",
         inplace=False,
     )