mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-08 08:47:03 +08:00
minor
Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
parent
a8e7071924
commit
c7f3e84b34
@ -142,7 +142,6 @@ def _apply_gumbel_kernel(
|
||||
seed = tl.load(seeds_ptr + req_idx)
|
||||
pos = tl.load(pos_ptr + req_idx)
|
||||
gumbel_seed = seed ^ (pos * 0x9E3779B9)
|
||||
gumbel_seed = gumbel_seed & 0xFFFFFFFF
|
||||
|
||||
block_id = tl.program_id(1)
|
||||
r_offset = block_id * BLOCK_SIZE + tl.arange(0, BLOCK_SIZE)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user