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_rope.py
+++ b/test/srt/cpu/test_rope.py
@@ -10,6 +10,8 @@ from sglang.srt.layers.rotary_embedding import (
 )
 from sglang.test.test_utils import CustomTestCase

+torch.manual_seed(0)
+

 class TestROPE(CustomTestCase):
    def test_deepseek_v2_rope(self):