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: if seeds is not None:
# Per-request seed. # Per-request seed.
assert pos is not None assert pos is not None
gumbel_seeds = seeds + pos gumbel_seeds = seeds ^ (pos * 0x9E3779B9)
else: else:
# Global seed. # Global seed.
assert pos is None assert pos is None