Skip to content
KernelIndex
Search⌘K

submission 512452

burtenshaw · python · License unknown

Use it

Vendorable · source mirrored · license unknownView source →

No package. Vendor the mirrored source: 12 lines, June 9 Researcher Reciprocity License v1.0.

submission_out.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-prefixsum-v2-512452?include=source"
interfacepython
Compatibility
measured onNVIDIA H100
declared hardwareNVIDIA H100
architecturessm_90
dtypesfp32

Benchmark evidence

1 measurement across 1 GPU, fastest first.

Operation / workload
Hardware
Latency
Rank
Observed
Inclusive prefix sumsuite of 11 cases
NVIDIA H100
1.02ms
#10 of 23
2026-03-02

Reported · How evidence levels are derived →

Source and license

sourceavailable
revision digestsha256:e8ce750f413ad38b5c7c955c02d1cebcfaeef5d48de3eb23bedd96ca0794e149
license declaredunknown
license concludedunknown
authorsburtenshaw
imported2026-08-15

Kernel source

submission_out.py12 lines
#!POPCORN leaderboard prefixsum_v2
#!POPCORN gpu A100

import torch
from task import input_t, output_t


def custom_kernel(data: input_t) -> output_t:
    x, y = data
    torch.cumsum(x, dim=0, out=y)
    return y

Source code from GPU Mode and the KernelBot dataset · June 9 Researcher Reciprocity License v1.0

Changes from previous submission

Against this author's previous submission submission 512437.

#!POPCORN leaderboard prefixsum_v2
- #!POPCORN gpu L4
+ #!POPCORN gpu A100
import torch
-
from task import input_t, output_t
- def _kernel(data: input_t) -> output_t:
+ def custom_kernel(data: input_t) -> output_t:
x, y = data
torch.cumsum(x, dim=0, out=y)
return y
-
-
- custom_kernel = torch.compile(_kernel, mode="reduce-overhead")

Best evidence level for this revision: reported

JSON