Antoni Baum 426ec4ec67 [1/n] Triton sampling kernel (#3186) ...

Co-authored-by: Roger Wang <136131678+ywang96@users.noreply.github.com>

2024-03-20 14:45:08 -07:00

..

test_beam_search.py

[FIX] Fix beam search test (#2930)

2024-02-20 14:37:39 -08:00

test_logprobs.py

Re-enable the 80 char line width limit (#3305)

2024-03-10 19:49:14 -07:00

test_rejection_sampler.py

Remove hardcoded device="cuda" to support more devices (#2503)

2024-02-01 15:46:39 -08:00

test_sampler.py

[1/n] Triton sampling kernel (#3186)

2024-03-20 14:45:08 -07:00

test_seeded_generate.py

Support per-request seed (#2514)

2024-02-21 11:47:00 -08:00