Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

The lookahead speculative decoding failed when applied to gpt2 0.1b. #2879

Closed
JeRainXiong opened this issue Mar 12, 2025 · 1 comment
Closed

Comments

@JeRainXiong
Copy link

Hi,
My environment is A10 22G, trtllm v0.16.0, and CUDA 12.6. When I use lookahead speculative decoding on gpt2 0.1B, I encounter this error. How can I fix it?

[TensorRT-LLM][ERROR] Assertion failed: No available XQA kernels are found for speculative decoding mode.
(/target/trtllm_1118/TensorRT-LLM/cpp/tensorrt 1lm/plugins/gptAttentionCommon/gptAttentionCommon.cpp:1745)
@JeRainXiong
Copy link
Author

I solve this problem by using TensorRT-LLM version: 0.18.0.dev2025031100

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant