PyTorch eager
Mamba2ReturnFinalState · d_head = 64 · n_heads = 8 · batch_size = 2048 · seq_length = 128 · fp32 · run 01a03f0e-f842…
●passedReported evidence
Not re-observed recently.
Primary measurement
9.18ms±0.02 · mean of 100
Rank 2 in its comparison group · source-native comparison · observed 2026-03-05
max ms 9.44 · min ms 9.11 · std ms 0.11 · mean ms 9.18
Identity
implementationPyTorch eager
projectPyTorch
revisionunknown
workloadd_head = 64 · n_heads = 8 · batch_size = 2048 · seq_length = 128 · fp32
comparison keysha256:1cf486996b1ab2bb…
sourceKernelBench baseline timings
external idH100_PCIe_LambdaLabs/torch/level3/49_Mamba2ReturnFinalState.py
sha256:fac5b88aa3f03d25d016001…
Correctness
Marked passed by the source; the correctness policy was not published.
Workload
d_head64
n_heads8
batch_size2048
seq_length128
input_1fp32 [2048, 128, 8, 64]
definition comparatornot_asserted
Measurements
latency · max9.44 ms · n=100
latency · mean9.18 ms · n=100
latency · min9.11 ms · n=100
latency · std110.0 µs
Protocol
harnessKernelBench timing scripts
timercuda_events
primaryStatisticmean
comparabilityFamilykernelbench_baseline_timing
Environment
gpuNVIDIA H100 (sm_90)
Artifacts
No artifacts published with this run.
Replications and notes
No attestations yet.
Community attestations. They never change the evidence level; only a KernelIndex-controlled rerun does.
Add a reproduction or note
Canonical manifest
Show manifestHide manifest
{
"run": {
"kind": "BenchmarkRun",
"spec": {
"status": "passed",
"timing": {
"samples": 100,
"latencyNs": {
"mean": 9180000,
"maximum": 9440000,
"minimum": 9110000,
"confidence95": [
9158440,
9201560
]
},
"primaryStatistic": "mean"
},
"observedAt": "2026-03-05T08:38:15.000Z",
"measurements": [
{
"unit": "ns",
"value": 110000,
"metric": "latency",
"statistic": "std"
}
],
"sourceNative": {
"source": "kernelbench",
"metrics": {
"max_ms": 9.44,
"min_ms": 9.11,
"std_ms": 0.11,
"mean_ms": 9.18
},
"benchmark": "H100_PCIe_LambdaLabs/baseline_time_torch.json",
"externalId": "H100_PCIe_LambdaLabs/torch/level3/49_Mamba2ReturnFinalState.py"
},
"protocolDigest": "sha256:547725b033a9911eae0c5ac362344a85aed824894ed4ab476af945b9bb491ea6",
"workloadDigest": "sha256:bab4ef5c86b034838b74468e3b6801f7d9c9d53660a76d51dda57cdea33526c7",
"environmentDigest": "sha256:89f9b28b06758f22a1df331fea58bfde2c9cc0f3627d1829509743ea538f0011",
"implementationDigest": "sha256:68d837fe8aae6933eebb13607729d7f065d93f6361b8a6957be8960bcdc04e83"
},
"metadata": {
"name": "kernelbench-h100-pcie-lambdalabs-torch-l3-49-mamba2returnfinalstate",
"title": "Mamba2ReturnFinalState · PyTorch eager · Lambda Labs"
},
"apiVersion": "kernelindex.dev/v1alpha1"
},
"protocol": {
"kind": "BenchmarkProtocol",
"spec": {
"harness": {
"name": "KernelBench timing scripts",
"repository": "https://github.com/ScalingIntelligence/KernelBench"
},
"measurement": {
"timer": "cuda_events",
"primaryStatistic": "mean"
},
"comparability": {
"notes": "Mean of 100 timed forward passes (CUDA events, warm-up excluded) of the reference module on fixed inputs; the 95% interval is the mean's, from the reported standard deviation. The upstream JSON records no torch or CUDA version. Comparable only within one problem and one timing host.",
"family": "kernelbench_baseline_timing"
}
},
"metadata": {
"name": "kernelbench-timing-v1",
"title": "KernelBench baseline timing"
},
"apiVersion": "kernelindex.dev/v1alpha1"
},
"environment": {
"kind": "ExecutionEnvironment",
"spec": {
"hardware": {
"vendor": "nvidia",
"product": "NVIDIA H100",
"formFactor": "PCIe",
"architecture": "sm_90"
},
"software": {}
},
"metadata": {
"name": "kernelbench-h100-pcie-lambdalabs",
"title": "KernelBench host · H100 PCIe (Lambda Labs)"
},
"apiVersion": "kernelindex.dev/v1alpha1"
}
}Cite this record (permalink, digest, access date)
Report an issue with this run
Published 2026-08-26 · KernelBench baseline timings · MITAll results for Mamba2ReturnFinalState →JSON