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_gemm.py
+++ b/test/srt/cpu/test_gemm.py
@@ -14,6 +14,8 @@ from utils import (

 from sglang.test.test_utils import CustomTestCase

+torch.manual_seed(0)
+

 class Mod(nn.Module):
    def __init__(self, input_channel, output_channel, has_bias):