add seed in CPU UTs to avoid flaky failure (#7333)

2025-06-19 10:12:14 +08:00
parent 20a503c7d1
commit 9179ea1595
10 changed files with 20 additions and 0 deletions
--- a/test/srt/cpu/test_topk.py
+++ b/test/srt/cpu/test_topk.py
@@ -13,6 +13,8 @@ from sglang.srt.layers.moe.topk import grouped_topk_gpu as native_grouped_topk
 from sglang.srt.models.llama4 import Llama4MoE
 from sglang.test.test_utils import CustomTestCase

+torch.manual_seed(0)
+

 # This is used by the Deepseek-V2 model
 class TestGroupedTopK(CustomTestCase):