submission 545192
rajesh0042 · python · License unknown
Use it
Vendorable · source mirrored · license unknownView source →
No package. Vendor the mirrored source: 13 lines, June 9 Researcher Reciprocity License v1.0.
prefixsum_v2.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-prefixsum-v2-545192?include=source"interfacepython
Compatibility
measured onNVIDIA B200
declared hardwareNVIDIA B200
architecturessm_100
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:495a1d6ce51476d1b6ebe24487c79a30ec3f4609d90c8e18c358b6fb805393a6
license declaredunknown
license concludedunknown
authorsrajesh0042
imported2026-08-15
Kernel source
prefixsum_v2.py13 lines
import os
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
import torch
from task import input_t, output_t
def custom_kernel(data: input_t) -> output_t:
data, output = data
# torch.cumsum is already very optimized
# Use contiguous data and out= parameter
torch.cumsum(data, dim=0, out=output)
return output
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 545075.
⋯ 5 unchanged linesdef custom_kernel(data: input_t) -> output_t:data, output = data+ # torch.cumsum is already very optimized+ # Use contiguous data and out= parametertorch.cumsum(data, dim=0, out=output)return output
Best evidence level for this revision: reported
JSON