Skip to content
KernelIndex
Search⌘K

Fused attention QK matmul scale mask softmax backward

39 eligible runs
attention

Backward pass for fused attention QK matmul with scaling, causal mask, and softmax. Computes gradients for query and key tensors through the chain: softmax_backward -> scale_backward -> matmul_backward.

Fastest reported
137.0µsmean · 71% of SOL
Amir M. Mir | SF TensorLicense unknown · unknown

Reported evidence · last observed 2026-08-22. Reported by source; not independently reproduced. No public source. License unknown.

No public sourceRun detail →

Current records

Workloadsuite of 16 cases · mean latencyhead_dim = 128 · num_heads = 12816 cases

Not measured on H100 for this workload. Challenges →

Source-native comparison · GPU NVIDIA B200 · Workload suite of 16 cases · mean latency · Protocol SOL-ExecBench evaluation stack v1.1 · mean · 27 results · last observed 2026-08-24Record history →
#
Implementation
Latency
vs #1
Trust
Observed
1
137.0µs71% SOL
1.00×
Reported · license unknown · no source
2026-08-22

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
2
138.5µs71% SOL
1.01×
Reported · license unknown · no source
2026-08-19

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
3
144.5µs69% SOL
1.05×
Reported · license unknown · no source
2026-07-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
4
145.5µs69% SOL
1.06×
Reported · license unknown · no source
2026-07-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
5
147.2µs69% SOL
1.07×
Reported · license unknown · no source
2026-08-21

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
6
151.1µs68% SOL
1.10×
Reported · license unknown · no source
2026-07-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
7
154.5µs67% SOL
1.13×
Reported · license unknown · no source
2026-08-15

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
8
154.6µs68% SOL
1.13×
Reported · license unknown · no source
2026-08-21

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
9
154.8µs67% SOL
1.13×
Reported · license unknown · no source
2026-08-15

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
10
155.0µs67% SOL
1.13×
Reported · license unknown · no source
2026-08-15

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
11
156.2µs67% SOL
1.14×
Reported · license unknown · no source
2026-07-13

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
12
156.8µs67% SOL
1.15×
Reported · license unknown · no source
2026-07-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
13
159.2µs67% SOL
1.16×
Reported · license unknown · no source
2026-07-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
14
160.5µs67% SOL
1.17×
Reported · license unknown · no source
2026-07-16

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
15
162.5µs66% SOL
1.19×
Reported · license unknown · no source
2026-08-24

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
16
166.4µs66% SOL
1.21×
Reported · license unknown · no source
2026-07-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
17
166.4µs66% SOL
1.21×
Reported · license unknown · no source
2026-08-23

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
18
176.0µs64% SOL
1.28×
Reported · license unknown · no source
2026-08-22

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
19
176.2µs64% SOL
1.29×
Reported · license unknown · no source
2026-08-18

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
20
180.3µs63% SOL
1.32×
Reported · license unknown · no source
2026-08-18

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
21
185.6µs62% SOL
1.36×
Reported · license unknown · no source
2026-08-13

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
22
187.5µs62% SOL
1.37×
Reported · license unknown · no source
2026-07-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
23
187.5µs62% SOL
1.37×
Reported · license unknown · no source
2026-08-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
24
221.5µs56% SOL
1.62×
Reported · license unknown · no source
2026-07-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
25
229.0µs56% SOL
1.67×
Reported · license unknown · no source
2026-07-23

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
26
264.6µs51% SOL
1.93×
Reported · license unknown · no source
2026-08-07

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →
27
1.22ms17% SOL
8.89×
Reported · license unknown · no source
2026-08-10

Measured exactly what you asked. Reported by source; not independently reproduced.

no sourcelicense unknownno install recipeRun detail →

Implementations

Implementation
Runtime
Best latency
Evidence
Availability
unknown
137.0µs
1.00×
Reported
license unknown · no source
unknown
138.5µs
1.01×
Reported
license unknown · no source
unknown
144.6µs
1.06×
Reported
license unknown · no source
unknown
149.8µs
1.09×
Reported
license unknown · no source
unknown
154.8µs
1.13×
Reported
license unknown · no source
unknown
154.5µs
1.13×
Reported
license unknown · no source
unknown
154.8µs
1.13×
Reported
license unknown · no source
unknown
155.0µs
1.13×
Reported
license unknown · no source
Show all 39 implementations ›
unknown
156.2µs
1.14×
Reported
license unknown · no source
unknown
160.5µs
1.17×
Reported
license unknown · no source
unknown
162.5µs
1.19×
Reported
license unknown · no source
unknown
178.5µs
1.30×
Reported
license unknown · no source
unknown
218.4µs
1.59×
Reported
license unknown · no source
unknown
229.0µs
1.67×
Reported
license unknown · no source
unknown
147.2µs
1.07×
Reported
license unknown · no source
unknown
151.1µs
1.10×
Reported
license unknown · no source
unknown
154.6µs
1.13×
Reported
license unknown · no source
unknown
156.8µs
1.15×
Reported
license unknown · no source
unknown
159.2µs
1.16×
Reported
license unknown · no source
unknown
165.1µs
1.21×
Reported
license unknown · no source
unknown
166.4µs
1.21×
Reported
license unknown · no source
unknown
166.2µs
1.21×
Reported
license unknown · no source
unknown
166.4µs
1.21×
Reported
license unknown · no source
unknown
176.0µs
1.28×
Reported
license unknown · no source
unknown
176.2µs
1.29×
Reported
license unknown · no source
unknown
176.0µs
1.29×
Reported
license unknown · no source
unknown
180.3µs
1.32×
Reported
license unknown · no source
unknown
185.6µs
1.36×
Reported
license unknown · no source
unknown
187.0µs
1.37×
Reported
license unknown · no source
unknown
187.5µs
1.37×
Reported
license unknown · no source
unknown
187.5µs
1.37×
Reported
license unknown · no source
unknown
221.5µs
1.62×
Reported
license unknown · no source
unknown
264.6µs
1.93×
Reported
license unknown · no source
unknown
270.1µs
1.97×
Reported
license unknown · no source
unknown
1.22ms
8.89×
Reported
license unknown · no source
unknown
1.24ms
9.08×
Reported
license unknown · no source
unknown
144.5µs
1.06×
Reported
license unknown · no source
unknown
144.5µs
1.05×
Reported
license unknown · no source
unknown
145.5µs
1.06×
Reported
license unknown · no source

Semantics

Inputs and outputs
grad_outputbf16 [batch_size, num_heads, seq_len_q, seq_len_k]
querybf16 [batch_size, num_heads, seq_len_q, head_dim]
keybf16 [batch_size, num_heads, seq_len_k, head_dim]
attn_weightsbf16 [batch_size, num_heads, seq_len_q, seq_len_k]
scalingfp32 scalar
grad_querybf16 [batch_size, num_heads, seq_len_q, head_dim]
grad_keybf16 [batch_size, num_heads, seq_len_k, head_dim]
Axes and behavior
head_dimconstant = 128
num_headsconstant = 128
seq_len_kvariable
seq_len_qvariable
batch_sizevariable
determinismunspecified
constraintsNo mutation or aliasing
Identity
alias060_fused_attention_qk_matmul_scale_mask_softmax_backwardmodelllama-3-1-405bsha256b90d813148c6…
Sources: NVIDIA SOL-ExecBench (2026-08-24)last observed 2026-08-24How records are decidedJSON