search

flashinfer-ai / flashinfer

FlashInfer: Kernel Library for LLM Serving
https://flashinfer.ai
Apache License 2.0
1.21k stars 111 forks source link

tests: add more unittests for logits cap #351

Closed yzh119 closed 3 months ago

yzh119 commented 3 months ago

add the case of logits_soft_case=1.0 for correctness tests. add batch decode/prefill tests.