EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Update seed in CPU UTs to avoid flaky failure with single test (#7544)

Browse Source
This commit is contained in:
YanbingJiang
2025-06-26 12:25:50 +08:00
committed by GitHub
parent 2390a2bc8d
commit 0e05fe8cf4
11 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/srt/cpu/test_qkv_proj_with_rope.py
View File

@@ -15,7 +15,7 @@ from sglang.test.test_utils import CustomTestCase
convert_weight_packed = torch.ops.sgl_kernel.convert_weight_packed
qkv_proj_with_rope = torch.ops.sgl_kernel.qkv_proj_with_rope
qkv_proj_with_rope_fused_weight = torch.ops.sgl_kernel.qkv_proj_with_rope_fused_weight
torch.manual_seed(0)
torch.manual_seed(1234)
# constants
kv_lora_rank = 512
qk_head_dim = 192