Skip to content

[RFE] πŸ¦— SDPA Support Matrix: Hopper (SM90) β€” open-source kernels wanted!Β #378

Description

@vedaanta

πŸ¦— Mission board: Hopper (SM90)

We're building open-source SDPA (flash attention) kernels for cuDNN frontend, and this is the coverage board for Hopper (SM90).

Right now the board is dark β€” every cell is up for grabs. Claim one, land it, and light it up. πŸ”¦

Legend

Symbol Meaning
βœ… Shipped & validated
🚧 In progress β€” someone's on it
❌ Open β€” claim it!
🚫 Not applicable on this architecture

The board

Feature Forward Backward
β€” Data types β€”
FP16 ❌ ❌
BF16 ❌ ❌
FP8 (E4M3 / E5M2) ❌ ❌
MXFP8 ❌ ❌
β€” Head dims β€”
d = 64 ❌ ❌
d = 128 ❌ ❌
d = 192/128 ❌ ❌
d = 256 ❌ ❌
d = 512 ❌ ❌
β€” Masks β€”
Causal ❌ ❌
Causal bottom-right (diagonal_alignment) ❌ ❌
Sliding window (SWA) ❌ ❌
Padding mask (per-batch seq lens) ❌ ❌
β€” Other β€”
GQA / MQA ❌ ❌
Attention bias (additive) ❌ ❌
ALiBi ❌ ❌
Attention sink ❌ ❌
THD / varlen (packed cu_seqlens layout) ❌ ❌
Dropout ❌ ❌
Paged attention (KV cache) ❌ ❌

How to play

  1. πŸ’¬ Comment below with the cell(s) you want to tackle.
  2. 🚧 We'll assign you and flip the cell to in progress.
  3. βœ… Land the PR with tests β€” cell lights up, scoreboard ticks. πŸŽ‰

Scoreboard: 0 / 40 cells lit.

Sister boards: ⚑ Ampere (SM80) #377 Β· πŸš€ Blackwell (SM100) #379 Β· πŸ’« Blackwell Ultra (SM103) #380 Β· ✨ Blackwell GeForce / DGX Spark (SM120/SM121) #381

Metadata

Metadata

Assignees

No one assigned

    Labels

    cat-featureRequests for new functionality, APIs, examples, or behavior improvements.mod-frontendcuDNN frontend APIs, operation graph construction, plans, and user-facing wrappers.open-todoOpen work that is ready to be picked up, scheduled, or prioritized.open-unassignedOpen work with no current assignee or clear owner.

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions