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_decode.py
+++ b/test/srt/cpu/test_decode.py
@@ -6,6 +6,8 @@ from torch.nn.functional import scaled_dot_product_attention

 from sglang.test.test_utils import CustomTestCase

+torch.manual_seed(0)
+

 class TestDecodeAttention(CustomTestCase):
    def _run_sdpa_forward_decode(