fix format issue

Signed-off-by: Yu Gong <[email protected]>

Author: yugong333

vllm/lora/punica_wrapper/punica_gpu.py

@@ -389,19 +389,19 @@ def add_lora_fused_moe(
             top_k_num,
             lora_ids,
             adapter_enabled,
-            shrink_config.get("BLOCK_SIZE_M", 64),
-            shrink_config.get("BLOCK_SIZE_N", 64),
-            shrink_config.get("BLOCK_SIZE_K", 32),
-            shrink_config.get("GROUP_SIZE_M", 8),
-            shrink_config.get("NUM_WARPS", 4),
-            shrink_config.get("NUM_STAGES", 3),
-            shrink_config.get("SPLIT_K", 1),
-            expand_config.get("BLOCK_SIZE_M", 64),
-            expand_config.get("BLOCK_SIZE_N", 64),
-            expand_config.get("BLOCK_SIZE_K", 64),
-            expand_config.get("GROUP_SIZE_M", 64),
-            expand_config.get("NUM_WARPS", 4),
-            expand_config.get("NUM_STAGES", 3),
-            expand_config.get("SPLIT_K", 1),
+            shrink_config.get("BLOCK_SIZE_M") or shrink_config.get("block_m") or 64,
+            shrink_config.get("BLOCK_SIZE_N") or shrink_config.get("block_n") or 64,
+            shrink_config.get("BLOCK_SIZE_K") or shrink_config.get("block_k") or 32,
+            shrink_config.get("GROUP_SIZE_M") or shrink_config.get("group_m") or 8,
+            shrink_config.get("NUM_WARPS") or shrink_config.get("num_warps") or 4,
+            shrink_config.get("NUM_STAGES") or shrink_config.get("num_stages") or 3,
+            shrink_config.get("SPLIT_K") or shrink_config.get("split_k") or 1,
+            expand_config.get("BLOCK_SIZE_M") or expand_config.get("block_m") or 64,
+            expand_config.get("BLOCK_SIZE_N") or expand_config.get("block_n") or 64,
+            expand_config.get("BLOCK_SIZE_K") or expand_config.get("block_k") or 64,
+            expand_config.get("GROUP_SIZE_M") or expand_config.get("group_m") or 64,
+            expand_config.get("NUM_WARPS") or expand_config.get("num_warps") or 4,
+            expand_config.get("NUM_STAGES") or expand_config.get("num_stages") or 3,
+            expand_config.get("SPLIT_K") or expand_config.get("split_k") or 1,
             mul_routed_weight,
         )