fix format issue

Signed-off-by: Yu Gong <yu3.gong@gmail.com>

Author: yugong333

vllm/lora/punica_wrapper/punica_gpu.py

@@ -382,19 +382,19 @@ def add_lora_fused_moe(
             num_tokens_post_padded,
             max_lora_rank,
             top_k_num,
-            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,
         )