frozenleaves/DeepSpeed
1
0
Fork 0
mirror of https://github.com/deepspeedai/DeepSpeed.git synced 2025-10-20 23:53:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Limit random seed range in tests (#7553)

Browse Source 
`pytest-randomly` often passes a large seed value to `set_random_seed`
and causes an error
([example](https://github.com/deepspeedai/DeepSpeed/actions/runs/17620450004/job/50064585974))
```
E ValueError: Seed must be between 0 and 2**32 - 1
```

This PR limits the range of seed values by taking a modulo.

Signed-off-by: Masahiro Tanaka <mtanaka@anyscale.com>
This commit is contained in:
Masahiro Tanaka
2025-09-10 10:45:37 -07:00
committed by GitHub
parent 8cbbbb539d
commit 0012ff6ea8
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
deepspeed/runtime/utils.py
View File

@ -103,8 +103,13 @@ def set_random_seed(seed):
import numpy
import random
random.seed(seed)
numpy.random.seed(seed)
torch.manual_seed(seed)
# pytest-randomly passes a too large seed
# `numpy.random.default_rng` could be a better approach, but it requires more changes to use rngs explicitly
# numpy.random accepts only 32-bit integers
numpy.random.seed(seed % (2**32))
# torch.manual_seed accepts only 64-bit integers
torch.manual_seed(seed % (2**63))
def is_model_parallel_parameter(p) -> bool: