"tests/python/pytorch/dataloading/test_dataloader.py" did not exist on "01bec4a31d1a3135af33ac69cefaf7ceecbaf7b3"
Unverified Commit 0e05fe8c authored by YanbingJiang's avatar YanbingJiang Committed by GitHub
Browse files

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

parent 2390a2bc
...@@ -8,7 +8,7 @@ from utils import SiluAndMul, precision ...@@ -8,7 +8,7 @@ from utils import SiluAndMul, precision
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
class TestActivation(CustomTestCase): class TestActivation(CustomTestCase):
......
...@@ -6,7 +6,7 @@ from torch.nn.functional import scaled_dot_product_attention ...@@ -6,7 +6,7 @@ from torch.nn.functional import scaled_dot_product_attention
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
class TestDecodeAttention(CustomTestCase): class TestDecodeAttention(CustomTestCase):
......
...@@ -6,7 +6,7 @@ from torch.nn.functional import scaled_dot_product_attention ...@@ -6,7 +6,7 @@ from torch.nn.functional import scaled_dot_product_attention
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
class TestExtendAttention(CustomTestCase): class TestExtendAttention(CustomTestCase):
......
...@@ -14,7 +14,7 @@ from utils import ( ...@@ -14,7 +14,7 @@ from utils import (
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
class Mod(nn.Module): class Mod(nn.Module):
......
...@@ -8,7 +8,7 @@ from utils import precision ...@@ -8,7 +8,7 @@ from utils import precision
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
class TestMLA(CustomTestCase): class TestMLA(CustomTestCase):
......
...@@ -8,7 +8,7 @@ import torch ...@@ -8,7 +8,7 @@ import torch
kernel = torch.ops.sgl_kernel kernel = torch.ops.sgl_kernel
torch.manual_seed(0) torch.manual_seed(1234)
from utils import ( from utils import (
BLOCK_K, BLOCK_K,
......
...@@ -8,7 +8,7 @@ from utils import make_non_contiguous, precision ...@@ -8,7 +8,7 @@ from utils import make_non_contiguous, precision
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
class TestNorm(CustomTestCase): class TestNorm(CustomTestCase):
......
...@@ -15,7 +15,7 @@ from sglang.test.test_utils import CustomTestCase ...@@ -15,7 +15,7 @@ from sglang.test.test_utils import CustomTestCase
convert_weight_packed = torch.ops.sgl_kernel.convert_weight_packed 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 = torch.ops.sgl_kernel.qkv_proj_with_rope
qkv_proj_with_rope_fused_weight = torch.ops.sgl_kernel.qkv_proj_with_rope_fused_weight 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 # constants
kv_lora_rank = 512 kv_lora_rank = 512
qk_head_dim = 192 qk_head_dim = 192
......
...@@ -10,7 +10,7 @@ from sglang.srt.layers.rotary_embedding import ( ...@@ -10,7 +10,7 @@ from sglang.srt.layers.rotary_embedding import (
) )
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
class TestROPE(CustomTestCase): class TestROPE(CustomTestCase):
......
...@@ -22,7 +22,7 @@ from utils import ( ...@@ -22,7 +22,7 @@ from utils import (
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
class TestSharedExpert(CustomTestCase): class TestSharedExpert(CustomTestCase):
......
...@@ -13,7 +13,7 @@ from sglang.srt.layers.moe.topk import grouped_topk_gpu as native_grouped_topk ...@@ -13,7 +13,7 @@ from sglang.srt.layers.moe.topk import grouped_topk_gpu as native_grouped_topk
from sglang.srt.models.llama4 import Llama4MoE from sglang.srt.models.llama4 import Llama4MoE
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
torch.manual_seed(0) torch.manual_seed(1234)
# This is used by the Deepseek-V2 model # This is used by the Deepseek-V2 model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment