Merged
fix: 合法上界种子与用例哈希组合后溢出导致 CPU 测试失败 #39
tianjunAdd on 29 days ago
fix: 合法上界种子与用例哈希组合后溢出导致 CPU 测试失败 #39
Merged
Total: 1 files changed+2-0
| @@ -14,6 +14,8 @@ def seed(request): | |||
| 14 | base = request.config.getoption('--seed') | 14 | base = request.config.getoption('--seed') |
| 15 | node_hash = int(hashlib.sha256(request.node.nodeid.encode()).hexdigest(), 16) % (2**31) | 15 | node_hash = int(hashlib.sha256(request.node.nodeid.encode()).hexdigest(), 16) % (2**31) |
| 16 | seed = base + node_hash | 16 | seed = base + node_hash |
| 17 | + if -(2**63) <= base <= 2**64 - 1 and seed > 2**64 - 1: | ||
| 18 | + seed %= 2**64 | ||
| 17 | random.seed(seed) | 19 | random.seed(seed) |
| 18 | torch.manual_seed(seed) | 20 | torch.manual_seed(seed) |
| 19 | return seed | 21 | return seed |