Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
Woosuk Kwon 2025-09-18 13:49:52 -07:00
parent a4962833f9
commit bc6463ac97

View File

@ -210,7 +210,7 @@ def gumbel_sample(
if seeds is not None:
# Per-request seed.
assert pos is not None
gumbel_seeds = seeds + pos
gumbel_seeds = seeds ^ (pos * 0x9E3779B9)
else:
# Global seed.
assert pos is None