submission 512313
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-512313?include=source"interfacepython
Compatibility
measured onNVIDIA A100
declared hardwareNVIDIA A100
architecturessm_80
dtypesfp32
Benchmark evidence
1 measurement across 1 GPU, fastest first.
Operation / workload
Hardware
Latency
Rank
Observed
Reported · How evidence levels are derived →
Source and license
sourceavailable
revision digestsha256:ee085ac13771125a78d0b33042b61e70a9eabd7a412ba437743625c709a3b4e4
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 512298.
⋯ 5 unchanged linesdef custom_kernel(data: input_t) -> output_t:- """- Reference implementation of inclusive prefix sum using PyTorch.- Args:- data: Input tensor to compute prefix sum on- Returns:- Tensor containing the inclusive prefix sum- """- data, output = data- output[...] = torch.cumsum(data, dim=0)- return output+ x, y = data+ torch.cumsum(x, dim=0, out=y)+ return y
Best evidence level for this revision: reported
JSON