SIGN IN SIGN UP

FlashInfer: Kernel Library for LLM Serving

0 0 188 Python

[chore] Add guard to blackwell GDN prefill (#3267)

<!-- .github/pull_request_template.md -->

## ๐Ÿ“Œ Description

Raise not implemented error when trying to use blackwell GDN prefill
kernel but cuda version earlier than 13

## ๐Ÿ” Related Issues

<!-- Link any related issues here -->

## ๐Ÿš€ Pull Request Checklist

Thank you for contributing to FlashInfer! Before we review your pull
request, please make sure the following items are complete.

### โœ… Pre-commit Checks

- [x] I have installed `pre-commit` by running `pip install pre-commit`
(or used your preferred method).
- [x] I have installed the hooks with `pre-commit install`.
- [x] I have run the hooks manually with `pre-commit run --all-files`
and fixed any reported issues.

> If you are unsure about how to set up `pre-commit`, see [the
pre-commit documentation](https://pre-commit.com/).

## ๐Ÿงช Tests

- [x] Tests have been added or updated as needed.
- [x] All tests are passing (`unittest`, etc.).

## Reviewer Notes

<!-- Optional: anything you'd like reviewers to focus on, concerns, etc.
-->


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

* **Bug Fixes**
* Improved Blackwell GPU compatibility with clearer validation for CUDA
major version.
* Now surfaces explicit, user-friendly errors when running on
unsupported CUDA versions (requires CUDA 13+).
* Provides distinct error feedback when required Blackwell prefill
support is unavailable, reducing ambiguous failures.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Signed-off-by: jiahanc <173873397+jiahanc@users.noreply.github.com>
J
Jiahan Chang (Cyrus) committed
9f4d16174e81dc818ca65f9eb930ec5fd8a9b395
Parent: bc39fbc
Committed by GitHub <noreply@github.com> on 5/12/2026, 1:02:16 AM