FlashAttention Max Sequence Length Illegal Access
When using variable length sequences in FlashAttention (e.g., passing `cu_seqlens`), if the actual sequence length in the data exceeds the `max_seqlen_k` or `max_seqlen_q` parameter passed to the kernel, the highly optimized block-wise memory loader reads outside the allocated KV-cache or attention mask bounds.
When using variable length sequences in FlashAttention (e.
- Symptom
RuntimeError: CUDA error: an illegal memory access was encountered- Root cause
- When using variable length sequences in FlashAttention (e.g., passing `cu_seqlens`), if the actual sequence length in the data exceeds the `max_seqlen_k` or `max_seqlen_q` parameter passed to the kernel, the highly optimized block-wise memory loader reads outside the allocated KV-cache or attention mask bounds.
- Recommended fix
- Dynamically compute max sequence length max_seqlen = int((cu_seqlens[1:] - cu_seqlens[:-1]).max()) Ensures the kernel is instantiated with the correct bounds for the current batch's actual sequence length.
- How Denpex helps
- Denpex investigates FlashAttention Max Sequence Length Illegal Access using the evidence you provide or your connected workload collects. Earlier rank, host or application evidence is needed to distinguish an initiating failure from a downstream report.
What this failure is
FlashAttention Max Sequence Length Illegal Access is a Memory failure seen during ML training runs. When using variable length sequences in FlashAttention (e.g., passing `cu_seqlens`), if the actual sequence length in the data exceeds the `max_seqlen_k` or `max_seqlen_q` parameter passed to the kernel, the highly optimized block-wise memory loader reads outside the allocated KV-cache or attention mask bounds. Common tags: Kernel Out Of Bounds.
Is this what broke your run? Paste your log.
You're reading about FlashAttention Max Sequence Length Illegal Access. Paste your own traceback and relevant evidence for an investigation of your workload, with a next action or a specific missing fact. A reference entry does not establish your cause. No account or card for the free diagnosis. Review data handling before submitting sensitive logs.
Before uploading, review cloud data handling and local options.
Upload your log file. Large files are processed in your browser. Only selected evidence is submitted when you run diagnosis.
Why it happens (the mechanism)
FlashAttention uses custom fused CUDA kernels that often bypass PyTorch's built-in bounds checking for performance. The resulting illegal memory access is silent until the context is corrupted.
What you'll observe
- RuntimeError: CUDA error: an illegal memory access was encountered
- flash_bwd_kernel
- flash_fwd_kernel
Common symptoms and what they mean
| Symptom | Why it happens |
|---|---|
| Training large language models with variable sequence lengths crashes unexpectedly. | When using variable length sequences in FlashAttention (e.g., passing `cu_seqlens`), if the actual sequence length in the data exceeds the `max_seqlen_k` or `max_seqlen_q` parameter passed to the kernel, the highly optimized block-wise memory loader reads outside the allocated KV-cache or attention mask bounds. |
| Only crashes on specific batches with exceptionally long sequences. | When using variable length sequences in FlashAttention (e.g., passing `cu_seqlens`), if the actual sequence length in the data exceeds the `max_seqlen_k` or `max_seqlen_q` parameter passed to the kernel, the highly optimized block-wise memory loader reads outside the allocated KV-cache or attention mask bounds. |
| Error happens deep inside the `flash_attn_func`. | When using variable length sequences in FlashAttention (e.g., passing `cu_seqlens`), if the actual sequence length in the data exceeds the `max_seqlen_k` or `max_seqlen_q` parameter passed to the kernel, the highly optimized block-wise memory loader reads outside the allocated KV-cache or attention mask bounds. |
Which systems are affected
- FlashAttention
- PyTorch
- CUDA
How to confirm this is the problem
Use this checklist to test the hypothesis against a small reproduction. No single line proves the root cause, so preserve the preceding events and compare one variable at a time.
- ✓Print the max difference in `cu_seqlens` for every batch before passing to FlashAttention.
- ✓Compare actual max sequence length to the statically allocated `max_seqlen` config.
Root cause
- When using variable length sequences in FlashAttention (e.g., passing `cu_seqlens`), if the actual sequence length in the data exceeds the `max_seqlen_k` or `max_seqlen_q` parameter passed to the kernel, the highly optimized block-wise memory loader reads outside the allocated KV-cache or attention mask bounds.
The fix and how to prevent it
Searchable error signature
RuntimeError: CUDA error: an illegal memory access was encounteredUse this text as a lookup key in logs and upstream issue trackers. It is not presented as a captured customer log. Confirm the cause from your own preceding events, versions, configuration and the cited references.
The fix and the prevention pattern
The root cause is on this page and stays free. A free account adds the exact remediation steps, saved history, and the fix on every entry in the encyclopedia.
Sign up free. Unlock the full analysisNo credit card. Daily allowance follows verified trust tier. Instant access.
Diagnose this failure in VS Code
Select the traceback or open the failed terminal, then run Denpex locally to see the initiating rank, collateral failures, exact fix, and verification command without uploading the log.
Install the free VS Code extensionDon't just read the fix, diagnose your run
The encyclopedia tells you what went wrong. Denpex tells you what went wrong in YOUR training run. With your logs, your config, and your stack.