Skip to content

Commit cd0a0ef

Browse files
committed
update seed
1 parent 4c0eabf commit cd0a0ef

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/model_optimizations/test_dsv3_fused_routing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -545,6 +545,7 @@ def test_dsv3_fused_routing_op(
545545
pytest.skip("Invalid configuration: exceeds kernel limits for n_group = 1")
546546

547547
# Generate random inputs
548+
torch.manual_seed(42)
548549
scores = torch.randn(num_tokens, num_experts, device="cuda", dtype=data_type)
549550
bias = torch.randn(num_experts, device="cuda", dtype=bias_type)
550551
routed_scaling_factor = 1.0

0 commit comments

Comments
 (0)