Skip to content

Port of: Handle None attn_bias in FusedSDPA tiling path- #1778 - #1784

Open
iboiko-habana wants to merge 1 commit into
mainfrom
iboiko/port1778
Open

Port of: Handle None attn_bias in FusedSDPA tiling path- #1778#1784
iboiko-habana wants to merge 1 commit into
mainfrom
iboiko/port1778

Conversation

@iboiko-habana

Copy link
Copy Markdown
Collaborator

No description provided.

Signed-off-by: Iryna Boiko <iboiko@habana.ai>
Copilot AI lite review requested due to automatic review settings September 7, 2026 09:25
@iboiko-habana
iboiko-habana deployed to pre-merge-approval September 7, 2026 09:25 — with GitHub Actions Active

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Warning

Copilot couldn't run its full agentic review because it didn't start before the timeout. Make sure your repository has a runner available, or add a copilot-code-review.yml file specifying one with the runs-on attribute. See the docs for more details.

Pull request overview

Updates the FusedSDPA q-tiling conditional to explicitly avoid entering the tiling path when attn_bias is None.

Changes:

  • Adds an explanatory comment about why tiling is unnecessary without a bias plane.
  • Expands the branching condition to treat attn_bias is None as a no-tiling case.

💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.

Comment on lines +484 to +485
# Without a bias there is no plane to tile; _fsdpa_num_q_tiles already returns 1 for that case.
if attn_bias is None or num_q_tiles == 1:
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

Successfully merging this pull request may close these issues.

3 participants