flashinfer-ai / flashinfer

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

test: fix unittest for group gemm #302

Closed yzh119 closed 3 weeks ago

yzh119 commented 3 weeks ago

The case of column_major=True and use_weight_indices=False is not properly handled.