search

flashinfer-ai / flashinfer

FlashInfer: Kernel Library for LLM Serving
https://flashinfer.ai
Apache License 2.0
822 stars 77 forks source link

sampling: fix alignment issue for vocab_size not divisible by vec_size #211

Closed yzh119 closed 2 months ago

yzh119 commented 2 months ago

Remove debug codes, and support different vec_size.