update test config

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

Author: dongfengy

tests/unittest/_torch/thop/parallel/test_moe.py

@@ -1171,8 +1171,8 @@ def test_no_autotune(self, num_tokens, hidden_size, intermediate_size,
                               use_topk_as_input=use_topk_as_input)
 
     @pytest.mark.parametrize("num_tokens", [1])
-    @pytest.mark.parametrize("hidden_size", [2048])
-    @pytest.mark.parametrize("intermediate_size", [2048])
+    @pytest.mark.parametrize("hidden_size", [512])
+    @pytest.mark.parametrize("intermediate_size", [512])
     @pytest.mark.parametrize(
         "routing_info",
         [