fix

Author: none

lightllm/common/fused_moe/moe_silu_and_mul_config.py

@@ -30,7 +30,10 @@ def try_to_get_best_config(
             config = finded_config[min(finded_config.keys(), key=lambda x: abs(int(x) - M))]
             return config
         else:
-            config = {"BLOCK_M": 16, "BLOCK_N": 128, "num_warps": 4, "NUM_STAGES": 5}
+            if M < 256:
+                config = {"BLOCK_M": 1, "BLOCK_N": 128, "num_warps": 1, "NUM_STAGES": 1}
+            else:
+                config = {"BLOCK_M": 16, "BLOCK_N": 128, "num_warps": 4, "NUM_STAGES": 5}
 
         return config