remove _grouped_size_compiled_for_decode_kernels (#3453)

This commit is contained in:
Yineng Zhang
2025-02-10 13:01:21 +08:00
committed by GitHub
parent 52a492a16e
commit 27c4c9cf52
2 changed files with 1 additions and 16 deletions

View File

@@ -22,7 +22,7 @@ pip install sgl-kernel --force-reinstall --no-deps
pip install -e "python[all]" --find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer/
```
Note: Please check the [FlashInfer installation doc](https://docs.flashinfer.ai/installation.html) to install the proper version according to your PyTorch and CUDA versions. If you meet with issue like **ImportError: cannot import name `_grouped_size_compiled_for_decode_kernels`**, installing FlashInfer with some older version like 0.1.6 instead of the latest version could solve it.
Note: Please check the [FlashInfer installation doc](https://docs.flashinfer.ai/installation.html) to install the proper version according to your PyTorch and CUDA versions.
Note: To AMD ROCm system with Instinct/MI GPUs, do following instead: