flashinfer-ai / flashinfer

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

bugfix: avoid potential illegal memory access #267

Closed yzh119 closed 1 month ago

yzh119 commented 1 month ago

Followup of #266, add guard to mask array access.