-
Notifications
You must be signed in to change notification settings - Fork 203
docs(troubleshooting): add Blackwell (sm_103) cuDNN SDPA failure and workaround #1491
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
base: main
Are you sure you want to change the base?
Changes from all commits
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
|
|
@@ -283,6 +283,31 @@ nvidia-smi | |||||||||||||
| nvidia-smi --query-gpu=driver_version --format=csv | ||||||||||||||
| ``` | ||||||||||||||
|
|
||||||||||||||
| **Blackwell GPUs (`sm_103`): generation fails at the first attention layer:** | ||||||||||||||
|
|
||||||||||||||
| On Blackwell-generation NVIDIA GPUs, PyTorch's default scaled dot-product attention (SDPA) can | ||||||||||||||
| select the cuDNN backend, which has no valid plan for this architecture: | ||||||||||||||
|
|
||||||||||||||
| ``` | ||||||||||||||
| cuDNN Frontend error: No valid execution plans built | ||||||||||||||
| ``` | ||||||||||||||
|
|
||||||||||||||
| With FLUX this hits the very first attention block in the UNet, so no image is ever produced. | ||||||||||||||
|
|
||||||||||||||
| ```bash | ||||||||||||||
| # Use a hand-written attention implementation that bypasses SDPA and cuDNN | ||||||||||||||
| python main.py --use-split-cross-attention | ||||||||||||||
|
|
||||||||||||||
| # Blackwell also requires a CUDA 12.8 or newer PyTorch build; | ||||||||||||||
| # builds compiled against cu121 will not run on these cards at all | ||||||||||||||
| ``` | ||||||||||||||
|
|
||||||||||||||
| <Note> | ||||||||||||||
| Disabling the cuDNN SDPA backend programmatically before startup — for example calling | ||||||||||||||
| `torch.backends.cuda.enable_cudnn_sdp(False)` from a `sitecustomize.py` — does not work, because | ||||||||||||||
| the flag is not retained through CUDA initialization. The launch flag is the reliable fix. | ||||||||||||||
|
Comment on lines
+306
to
+308
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 📐 Maintainability & Code Quality | 🟡 Minor | ⚡ Quick win Replace the em dash with direct sentences. Rewrite this note without the em dash: Proposed wording-Disabling the cuDNN SDPA backend programmatically before startup — for example calling
-`torch.backends.cuda.enable_cudnn_sdp(False)` from a `sitecustomize.py` — does not work, because
-the flag is not retained through CUDA initialization. The launch flag is the reliable fix.
+Disabling the cuDNN SDPA backend programmatically before startup does not work. For example,
+calling `torch.backends.cuda.enable_cudnn_sdp(False)` from a `sitecustomize.py` does not persist
+through CUDA initialization. The launch flag is the reliable fix.As per coding guidelines, English MDX must avoid em dashes and prefer short, direct sentences. 📝 Committable suggestion
Suggested change
🤖 Prompt for AI AgentsSource: Coding guidelines |
||||||||||||||
| </Note> | ||||||||||||||
|
|
||||||||||||||
| ### AMD GPU Issues | ||||||||||||||
|
|
||||||||||||||
| **ROCm support (Linux only):** | ||||||||||||||
|
|
||||||||||||||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🎯 Functional Correctness | 🟡 Minor | ⚡ Quick win
Clarify the tested environment and support boundary.
Please include the exact tested GPU model and software versions: B300 SXM6, Ubuntu 24.04, CUDA 13.0 driver, PyTorch 2.11.0+cu128, and ComfyUI. Limit the result explicitly to that tested setup. Qualify any
cu121or CUDA support claim because nativesm_103compiler support starts with CUDA 12.9, while CUDA 12.8 may depend on PTX compatibility. Avoid implying that all Blackwell cards or configurations are affected unless verified.📍 Affects 1 file
troubleshooting/overview.mdx#L286-L303(this comment)troubleshooting/overview.mdx#L301-L303🤖 Prompt for AI Agents